Skip to content

Commit e99cfbc

Browse files
PR tensorflow#510: Fixing dataset directory path and tensorflow 2.0 fit module
Please approve this CL. It will be submitted automatically, and its GitHub pull request will be marked as merged. Imported from GitHub PR tensorflow#510 Existing code had a issue for dataset directory and also the current version of tensorflow 2.0 deprecated fit_generator and introduced fit method for the same functionality. Copybara import of the project: - b3ac75f Fixing dataset directory path and tensorflow 2.0 fit modu... by sablevinayak <[email protected]> - 8af8226 Merge b3ac75f into 16a5a... by Vinayak Sable <[email protected]> PiperOrigin-RevId: 757085715
1 parent 7103791 commit e99cfbc

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)