Преглед на файлове

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.