Change `steps_per_loop` check as it's a integer flag

PiperOrigin-RevId: 321398549

Change `steps_per_loop` check as it's a integer flag
PiperOrigin-RevId: 321398549
809e19af · Jin Young Sohn · A. Unique TensorFlower · 8b510f95 · 809e19af
Commit 809e19af authored Jul 15, 2020 by Jin Young Sohn Committed by A. Unique TensorFlower Jul 15, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py ...n/image_classification/resnet/resnet_ctl_imagenet_main.py +1 -1

No files found.
--- a/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
+++ b/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
@@ -125,7 +125,7 @@ def run(flags_obj):

  per_epoch_steps, train_epochs, eval_steps = get_num_train_iterations(
      flags_obj)
-  if flags_obj.steps_per_loop is None:
+  if not flags_obj.steps_per_loop:
    steps_per_loop = per_epoch_steps
  elif flags_obj.steps_per_loop > per_epoch_steps:
    steps_per_loop = per_epoch_steps