Notebooks

Lightning-Universe · Mar 29, 2021 · fb46458 · fb46458
1 parent e72d96d
commit fb46458
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flash_notebooks/image_classification.ipynb b/flash_notebooks/image_classification.ipynb
@@ -134,7 +134,7 @@
    "source": [
     "datamodule = ImageClassificationData.from_folders(\n",
     "    train_folder=\"data/hymenoptera_data/train/\",\n",
-    "    valid_folder=\"data/hymenoptera_data/val/\",\n",
+    "    val_folder=\"data/hymenoptera_data/val/\",\n",
     "    test_folder=\"data/hymenoptera_data/test/\",\n",
     ")"
    ]

diff --git a/flash_notebooks/text_classification.ipynb b/flash_notebooks/text_classification.ipynb
@@ -111,7 +111,7 @@
    "source": [
     "datamodule = TextClassificationData.from_files(\n",
     "    train_file=\"data/imdb/train.csv\",\n",
-    "    valid_file=\"data/imdb/valid.csv\",\n",
+    "    val_file=\"data/imdb/valid.csv\",\n",
     "    test_file=\"data/imdb/test.csv\",\n",
     "    input=\"review\",\n",
     "    target=\"sentiment\",\n",