소스 검색

Print warning on checkpoint not found

Eric Wang 3 년 전
부모
커밋
683810b4a1
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      finetune.py

+ 2 - 0
finetune.py

@@ -155,6 +155,8 @@ def train(
             print(f"Restarting from {checkpoint_name}")
             adapters_weights = torch.load(checkpoint_name)
             model = set_peft_model_state_dict(model, adapters_weights)
+        else:
+            print(f"Checkpoint {checkpoint_name} not found")
 
     model.print_trainable_parameters()  # Be more transparent about the % of trainable params.