diff --git a/flash_notebooks/image_classification.ipynb b/flash_notebooks/image_classification.ipynb index 4bf6ba2aae..cb25a201ca 100644 --- 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 index 9ad20120ee..f238c09b60 100644 --- 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",