From f87dd910f871647c882574a2c4158b1ec95625ed Mon Sep 17 00:00:00 2001 From: travis-the-dragon Date: Fri, 9 Aug 2024 10:26:50 -0600 Subject: [PATCH] Updated log_model reference to match torch.save. --- colabs/intro/Intro_to_Weights_&_Biases.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colabs/intro/Intro_to_Weights_&_Biases.ipynb b/colabs/intro/Intro_to_Weights_&_Biases.ipynb index 8725dbcc..1161e41b 100644 --- a/colabs/intro/Intro_to_Weights_&_Biases.ipynb +++ b/colabs/intro/Intro_to_Weights_&_Biases.ipynb @@ -317,7 +317,7 @@ "\n", " # 🐝 Save the model checkpoint to wandb\n", " torch.save(model, \"my_model.pt\")\n", - " wandb.log_model(\"./mnist_model.pt\", \"my_mnist_model\", aliases=[f\"epoch-{epoch+1}_dropout-{round(wandb.config.dropout, 4)}\"])\n", + " wandb.log_model(\"./my_model.pt\", \"my_mnist_model\", aliases=[f\"epoch-{epoch+1}_dropout-{round(wandb.config.dropout, 4)}\"])\n", "\n", " print(f\"Epoch: {epoch+1}, Train Loss: {train_loss:.3f}, Valid Loss: {val_loss:3f}, Accuracy: {accuracy:.2f}\")\n", "\n",