From d47a39ee021e89ab347857f0d8e5908de8f41c19 Mon Sep 17 00:00:00 2001 From: Jirka B Date: Wed, 20 Nov 2024 14:17:30 +0100 Subject: [PATCH] example: simple upload/download demo --- examples/__init__.py | 0 examples/demo-upload-download.py | 25 +++++++++++++++++++ .../{train_callback.py => train-callback.py} | 0 examples/{train_resume.py => train-resume.py} | 0 examples/{train_simple.py => train-simple.py} | 0 5 files changed, 25 insertions(+) delete mode 100644 examples/__init__.py create mode 100644 examples/demo-upload-download.py rename examples/{train_callback.py => train-callback.py} (100%) rename examples/{train_resume.py => train-resume.py} (100%) rename examples/{train_simple.py => train-simple.py} (100%) diff --git a/examples/__init__.py b/examples/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/examples/demo-upload-download.py b/examples/demo-upload-download.py new file mode 100644 index 0000000..e00756e --- /dev/null +++ b/examples/demo-upload-download.py @@ -0,0 +1,25 @@ +import litmodels +import torch +from lightning.pytorch.demos.boring_classes import BoringModel + +if __name__ == "__main__": + # Define your model + model = BoringModel() + + # Save the model's state dictionary + torch.save(model.state_dict(), "./boring-checkpoint.pt") + + # Upload the model checkpoint + litmodels.upload_model( + "./boring-checkpoint.pt", + "jirka/kaggle/boring-model", + ) + + # Download the model checkpoint + model_path = litmodels.download_model("jirka/kaggle/boring-model", download_dir="./my-models") + print(f"Model downloaded to {model_path}") + + # Load the model checkpoint + loaded_model = BoringModel() + loaded_model.load_state_dict(torch.load("./boring-checkpoint.pt")) + print(loaded_model) diff --git a/examples/train_callback.py b/examples/train-callback.py similarity index 100% rename from examples/train_callback.py rename to examples/train-callback.py diff --git a/examples/train_resume.py b/examples/train-resume.py similarity index 100% rename from examples/train_resume.py rename to examples/train-resume.py diff --git a/examples/train_simple.py b/examples/train-simple.py similarity index 100% rename from examples/train_simple.py rename to examples/train-simple.py