Skip to content

Commit b3ac75f

Browse files
committed
Fixing dataset directory path and tensorflow 2.0 fit module instead of fit_generator
1 parent 5c3457a commit b3ac75f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

courses/udacity_intro_to_tensorflow_for_deep_learning/l05c01_dogs_vs_cats_without_augmentation.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,7 @@
230230
},
231231
"outputs": [],
232232
"source": [
233-
"base_dir = os.path.join(os.path.dirname(zip_dir), 'cats_and_dogs_filtered')\n",
233+
"base_dir = os.path.join(zip_dir, 'cats_and_dogs_filtered')\n",
234234
"train_dir = os.path.join(base_dir, 'train')\n",
235235
"validation_dir = os.path.join(base_dir, 'validation')\n",
236236
"\n",
@@ -591,7 +591,7 @@
591591
"outputs": [],
592592
"source": [
593593
"EPOCHS = 100\n",
594-
"history = model.fit_generator(\n",
594+
"history = model.fit(\n",
595595
" train_data_gen,\n",
596596
" steps_per_epoch=int(np.ceil(total_train / float(BATCH_SIZE))),\n",
597597
" epochs=EPOCHS,\n",

0 commit comments

Comments
 (0)