"...autoencoder/autoencoder_models/VariationalAutoencoder.py" did not exist on "70702f79c7817b09fb87fef7729478af58532870"
Commit 81cf6368 authored by Allen Wang's avatar Allen Wang Committed by A. Unique TensorFlower
Browse files

Change steps_per_loop to `None` by default.

PiperOrigin-RevId: 321855387
parent 9c314a03
...@@ -255,7 +255,7 @@ def define_keras_flags( ...@@ -255,7 +255,7 @@ def define_keras_flags(
name='tpu', default='', help='TPU address to connect to.') name='tpu', default='', help='TPU address to connect to.')
flags.DEFINE_integer( flags.DEFINE_integer(
name='steps_per_loop', name='steps_per_loop',
default=500, default=None,
help='Number of steps per training loop. Only training step happens ' help='Number of steps per training loop. Only training step happens '
'inside the loop. Callbacks will not be called inside. Will be capped at ' 'inside the loop. Callbacks will not be called inside. Will be capped at '
'steps per epoch.') 'steps per epoch.')
......
...@@ -125,7 +125,7 @@ def run(flags_obj): ...@@ -125,7 +125,7 @@ def run(flags_obj):
per_epoch_steps, train_epochs, eval_steps = get_num_train_iterations( per_epoch_steps, train_epochs, eval_steps = get_num_train_iterations(
flags_obj) flags_obj)
if not flags_obj.steps_per_loop: if flags_obj.steps_per_loop is None:
steps_per_loop = per_epoch_steps steps_per_loop = per_epoch_steps
elif flags_obj.steps_per_loop > per_epoch_steps: elif flags_obj.steps_per_loop > per_epoch_steps:
steps_per_loop = per_epoch_steps steps_per_loop = per_epoch_steps
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment