Internal change

PiperOrigin-RevId: 300247220

Internal change
PiperOrigin-RevId: 300247220
7708847e · A. Unique TensorFlower · f7843c44 · 7708847e
Commit 7708847e authored Mar 10, 2020 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 2 deletions

official/benchmark/models/resnet_cifar_main.py official/benchmark/models/resnet_cifar_main.py +0 -2

No files found.
--- a/official/benchmark/models/resnet_cifar_main.py
+++ b/official/benchmark/models/resnet_cifar_main.py
@@ -174,7 +174,6 @@ def run(flags_obj):
      is_training=True,
      data_dir=flags_obj.data_dir,
      batch_size=flags_obj.batch_size,
-      num_epochs=flags_obj.train_epochs,
      parse_record_fn=cifar_preprocessing.parse_record,
      datasets_num_private_threads=flags_obj.datasets_num_private_threads,
      dtype=dtype,
@@ -189,7 +188,6 @@ def run(flags_obj):
        is_training=False,
        data_dir=flags_obj.data_dir,
        batch_size=flags_obj.batch_size,
-        num_epochs=flags_obj.train_epochs,
        parse_record_fn=cifar_preprocessing.parse_record)

  steps_per_epoch = (