lightning-ai/jirka/lit-boring

Author: Borda

examples/demo-upload-download.py

@@ -12,11 +12,11 @@
     # Upload the model checkpoint
     litmodels.upload_model(
         "./boring-checkpoint.pt",
-        "jirka/kaggle/boring-model",
+        "lightning-ai/jirka/lit-boring-model",
     )
 
     # Download the model checkpoint
-    model_path = litmodels.download_model("jirka/kaggle/boring-model", download_dir="./my-models")
+    model_path = litmodels.download_model("lightning-ai/jirka/lit-boring-model", download_dir="./my-models")
     print(f"Model downloaded to {model_path}")
 
     # Load the model checkpoint

examples/resume-lightning-training.py

@@ -2,19 +2,21 @@
 This example demonstrates how to resume training of a model using the `download_model` function.
 """
 
+import os
+
 from lightning import Trainer
 from lightning.pytorch.demos.boring_classes import BoringModel
 from litmodels import download_model
 
 # Define the model name - this should be unique to your model
 # The format is <organization>/<teamspace>/<model-name>:<model-version>
-MY_MODEL_NAME = "jirka/kaggle/lit-auto-encoder-callback:latest"
+MY_MODEL_NAME = "lightning-ai/jirka/lit-boring-callback:latest"
 
 
 if __name__ == "__main__":
-    model_path = download_model(name=MY_MODEL_NAME, download_dir="my_models")
+    model_files = download_model(name=MY_MODEL_NAME, download_dir="my_models")
+    model_path = os.path.join("my_models", model_files[0])
     print(f"model: {model_path}")
-    # autoencoder = LitAutoEncoder.load_from_checkpoint(checkpoint_path=model_path)
 
     trainer = Trainer(max_epochs=4)
     trainer.fit(

examples/train-model-and-simple-save.py

@@ -8,7 +8,7 @@
 
 # Define the model name - this should be unique to your model
 # The format is <organization>/<teamspace>/<model-name>
-MY_MODEL_NAME = "jirka/kaggle/lit-auto-encoder-simple"
+MY_MODEL_NAME = "lightning-ai/jirka/lit-boring-simple"
 
 
 if __name__ == "__main__":

examples/train-model-with-lightning-callback.py

@@ -8,7 +8,7 @@
 
 # Define the model name - this should be unique to your model
 # The format is <organization>/<teamspace>/<model-name>
-MY_MODEL_NAME = "lightning-ai/jirka/lit-auto-encoder-callback"
+MY_MODEL_NAME = "lightning-ai/jirka/lit-boring-callback"
 
 
 if __name__ == "__main__":