Remove unused num_epochs param in resnet_imagenet_main.py.

PiperOrigin-RevId: 299998039

Remove unused num_epochs param in resnet_imagenet_main.py.
PiperOrigin-RevId: 299998039
15eab100 · Rick Chao · A. Unique TensorFlower · f3a29bdd · 15eab100
Commit 15eab100 authored Mar 09, 2020 by Rick Chao Committed by A. Unique TensorFlower Mar 09, 2020
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 2 deletions

official/vision/image_classification/resnet_imagenet_main.py official/vision/image_classification/resnet_imagenet_main.py +0 -2

No files found.
--- a/official/vision/image_classification/resnet_imagenet_main.py
+++ b/official/vision/image_classification/resnet_imagenet_main.py
@@ -126,7 +126,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=imagenet_preprocessing.get_parse_record_fn(
          use_keras_image_data_format=use_keras_image_data_format),
      datasets_num_private_threads=flags_obj.datasets_num_private_threads,
@@ -142,7 +141,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=imagenet_preprocessing.get_parse_record_fn(
            use_keras_image_data_format=use_keras_image_data_format),
        dtype=dtype,