Internal change

PiperOrigin-RevId: 414542003

Internal change
PiperOrigin-RevId: 414542003
5c5424cf · A. Unique TensorFlower · ca67e4ec · 5c5424cf
Commit 5c5424cf authored Dec 06, 2021 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

official/legacy/image_classification/resnet/resnet_runnable.py ...ial/legacy/image_classification/resnet/resnet_runnable.py +2 -1

No files found.
--- a/official/legacy/image_classification/resnet/resnet_runnable.py
+++ b/official/legacy/image_classification/resnet/resnet_runnable.py
@@ -99,7 +99,8 @@ class ResnetRunnable(orbit.StandardTrainer, orbit.StandardEvaluator):
        datasets_num_private_threads=self.flags_obj
        .datasets_num_private_threads,
        dtype=self.dtype,
-        drop_remainder=True)
+        drop_remainder=True,
+        training_dataset_cache=self.flags_obj.training_dataset_cache)
    orbit.StandardTrainer.__init__(
        self,
        train_dataset,