udpated notebook docs exmaple

neptune-ai · Oct 9, 2019 · e94fe90 · e94fe90
1 parent 93e9a16
commit e94fe90
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/docs/examples/monitor_fastai.ipynb b/docs/examples/monitor_fastai.ipynb
@@ -104,17 +104,12 @@
     "\n",
     "neptune.init(project_qualified_name='USER_NAME/PROJECT_NAME')\n",
     "\n",
-    "mnist = untar_data(URLs.MNIST_TINY)\n",
-    "tfms = get_transforms(do_flip=False)\n",
+    "path = untar_data(URLs.MNIST_TINY)\n",
     "\n",
-    "data = (ImageItemList.from_folder(mnist)\n",
-    "        .split_by_folder()\n",
-    "        .label_from_folder()\n",
-    "        .transform(tfms, size=32)\n",
-    "        .databunch()\n",
-    "        .normalize(imagenet_stats))\n",
+    "data = ImageDataBunch.from_folder(path, ds_tfms=(rand_pad(2, 28), []), bs=64)\n",
+    "data.normalize(imagenet_stats)\n",
     "\n",
-    "learn = create_cnn(data, models.resnet18, metrics=accuracy)\n",
+    "learn = cnn_learner(data, models.resnet18, metrics=accuracy)\n",
     "learn.lr_find()\n",
     "learn.recorder.plot()\n",
     "    \n",