examples: fix argument

Borda · Borda · commit 3df3e21e7c12 · 2024-12-10T23:58:04.000+09:00
diff --git a/examples/train-callback.py b/examples/train-callback.py
@@ -15,7 +15,7 @@ def on_train_epoch_end(self, trainer, pl_module):
         best_model_path = trainer.checkpoint_callback.best_model_path
         if best_model_path:
             print(f"Uploading model: {best_model_path}")
-            upload_model(path=best_model_path, name=MY_MODEL_NAME)
+            upload_model(model=best_model_path, name=MY_MODEL_NAME)
 
 
 if __name__ == "__main__":
diff --git a/examples/train-simple.py b/examples/train-simple.py
@@ -35,4 +35,4 @@
         data.DataLoader(val, batch_size=256),
     )
     print(f"last: {vars(checkpoint_callback)}")
-    upload_model(path=checkpoint_callback.last_model_path, name=MY_MODEL_NAME)
+    upload_model(model=checkpoint_callback.last_model_path, name=MY_MODEL_NAME)

Original file line number	Diff line number	Diff line change
`@@ -35,4 +35,4 @@`
`35`	`35`	`data.DataLoader(val, batch_size=256),`
`36`	`36`	`)`
`37`	`37`	`print(f"last: {vars(checkpoint_callback)}")`
`38`		`- upload_model(path=checkpoint_callback.last_model_path, name=MY_MODEL_NAME)`
	`38`	`+ upload_model(model=checkpoint_callback.last_model_path, name=MY_MODEL_NAME)`