tweak final save name to not have the step number

johnowhitaker · web-flow · commit 388d0043edee · 2022-12-07T15:56:51.000+02:00
diff --git a/unit2/finetune_model.py b/unit2/finetune_model.py
@@ -114,7 +114,7 @@ def transform(examples):
         scheduler.step()
 
     # Save the pipeline one last time
-    image_pipe.save_pretrained(model_save_name+f'step_{step+1}')
+    image_pipe.save_pretrained(model_save_name)
     
     # Wrap up the run
-    wandb.finish()
+    wandb.finish()
