update in code to acept repeat

aws · May 6, 2024 · 7b7aa48 · 7b7aa48
1 parent 34c04b8
commit 7b7aa48
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...arning/object_detection_with_tensorflow_and_tfrecords/2_object_detection_train_eval.ipynb b/...arning/object_detection_with_tensorflow_and_tfrecords/2_object_detection_train_eval.ipynb
@@ -472,6 +472,8 @@
     "# Shuffle the prepared dataset\n",
     "sample_ds = parsed_dataset_sample.shuffle(BATCH_SIZE * 4)\n",
     "\n",
+    "sample_ds = sample_ds.repeat()\n",
+    "\n",
     "# Get a sample from the shuffled dataset\n",
     "data_not_augmented = next(iter(sample_ds.take(1)))\n",
     "\n",