examples

Author: Borda

examples/demo-upload-download.py

@@ -10,7 +10,7 @@
     torch.save(model.state_dict(), "./boring-checkpoint.pt")
 
     # Upload the model checkpoint
-    litmodels.upload_model_files(
+    litmodels.upload_model(
         "./boring-checkpoint.pt",
         "jirka/kaggle/boring-model",
     )

examples/train-callback.py

@@ -1,7 +1,7 @@
 import torch.utils.data as data
 import torchvision as tv
 from lightning import Callback, Trainer
-from litmodels import upload_model_files
+from litmodels import upload_model
 from sample_model import LitAutoEncoder
 
 
@@ -11,7 +11,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_files(path=best_model_path, name="jirka/kaggle/lit-auto-encoder-callback")
+            upload_model(path=best_model_path, name="jirka/kaggle/lit-auto-encoder-callback")
 
 
 if __name__ == "__main__":

examples/train-simple.py

@@ -2,7 +2,7 @@
 import torchvision as tv
 from lightning import Trainer
 from lightning.pytorch.callbacks import ModelCheckpoint
-from litmodels import upload_model_files
+from litmodels import upload_model
 from sample_model import LitAutoEncoder
 
 if __name__ == "__main__":
@@ -30,4 +30,4 @@
         data.DataLoader(val, batch_size=256),
     )
     print(f"last: {vars(checkpoint_callback)}")
-    upload_model_files(path=checkpoint_callback.last_model_path, name="jirka/kaggle/lit-auto-encoder-simple")
+    upload_model(path=checkpoint_callback.last_model_path, name="jirka/kaggle/lit-auto-encoder-simple")