update logo

NVIDIA-Merlin · Jul 6, 2023 · b54569b · b54569b
1 parent 09b4688
commit b54569b
Showing 1 changed file with 18 additions and 34 deletions.
diff --git a/examples/pytorch/01-Getting-started.ipynb b/examples/pytorch/01-Getting-started.ipynb
@@ -31,7 +31,7 @@
    "id": "23d9bf34",
    "metadata": {},
    "source": [
-    "<img src=\"https://developer.download.nvidia.com/notebooks/dlsw-notebooks/merlin_models_01-getting-started/nvidia_logo.png\" style=\"width: 90px; float: right;\">\n",
+    "<img src=\"https://developer.download.nvidia.com/notebooks/dlsw%20notebooks/merlin_models_pytorch_01-getting-started/nvidia_logo.png\" style=\"width: 90px; float: right;\">\n",
     "\n",
     "# Getting Started with Merlin Models: Develop a Model for MovieLens using the PyTorch API\n",
     "\n",
@@ -189,7 +189,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Epoch 0: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 782/782 [00:07<00:00, 100.68it/s, v_num=7]"
+      "Epoch 0: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████| 782/782 [00:07<00:00, 101.54it/s, v_num=10]"
      ]
     },
     {
@@ -203,16 +203,16 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Epoch 0: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 782/782 [00:07<00:00, 100.35it/s, v_num=7]\n"
+      "Epoch 0: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████| 782/782 [00:07<00:00, 101.21it/s, v_num=10]\n"
      ]
     }
    ],
    "source": [
     "trainer = pl.Trainer(max_epochs=1)\n",
+    "train_loader = Loader(train, batch_size=1024)\n",
     "\n",
-    "with Loader(train, batch_size=1024) as loader:\n",
-    "    model.initialize(loader)\n",
-    "    trainer.fit(model, loader)"
+    "model.initialize(train_loader)\n",
+    "trainer.fit(model, train_loader)"
    ]
   },
   {
@@ -240,7 +240,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Validation DataLoader 0:   7%|██████████████████████▍                                                                                                                                                                                                                                                                                                   | 14/196 [00:00<00:01, 157.25it/s]"
+      "Validation DataLoader 0:   7%|██████▌                                                                                            | 13/196 [00:00<00:01, 161.86it/s]"
      ]
     },
     {
@@ -255,15 +255,15 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Validation DataLoader 0: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 196/196 [00:01<00:00, 164.10it/s]\n",
+      "Validation DataLoader 0: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 196/196 [00:01<00:00, 165.71it/s]\n",
       "────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────\n",
       "     Validate metric           DataLoader 0\n",
       "────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────\n",
-      "  valid_binary_accuracy     0.7193925380706787\n",
-      "   valid_binary_auroc       0.7811704874038696\n",
-      " valid_binary_precision     0.7327381372451782\n",
-      "   valid_binary_recall      0.8074849843978882\n",
-      "       valid_loss           0.5524654984474182\n",
+      "   val_binary_accuracy      0.7193425297737122\n",
+      "    val_binary_auroc        0.7803523540496826\n",
+      "  val_binary_precision      0.7274115681648254\n",
+      "    val_binary_recall       0.8201844692230225\n",
+      "        val_loss            0.5525734424591064\n",
       "────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────\n"
      ]
     },
@@ -277,8 +277,8 @@
     }
    ],
    "source": [
-    "with Loader(valid, batch_size=1024) as loader:\n",
-    "    metrics = trainer.validate(model, loader)"
+    "val_loader = Loader(valid, batch_size=1024)\n",
+    "metrics = trainer.validate(model, val_loader)"
    ]
   },
   {
@@ -306,28 +306,12 @@
     ")\n",
     "\n",
     "trainer = pl.Trainer(max_epochs=1)\n",
+    "train_loader = Loader(train, batch_size=1024)\n",
     "\n",
-    "with Loader(train, batch_size=1024) as loader:\n",
-    "    model.initialize(loader)\n",
-    "    trainer.fit(model, loader)\n",
+    "model.initialize(train_loader)\n",
+    "trainer.fit(model, train_loader)\n",
     "```"
    ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "c9180e84",
-   "metadata": {},
-   "source": [
-    "## Next steps"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "b9ba3102",
-   "metadata": {},
-   "source": [
-    "In the next example notebooks, we will show how the integration with NVTabular and how to explore different recommender models."
-   ]
   }
  ],
  "metadata": {