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