Updated log_model reference to match torch.save. (#562)

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",