@@ -94,9 +94,9 @@ class Trainer(object):
9494 """
9595
9696 def __init__ (
97- self , training_dataset , build_training_func , optimizer , optimizer_args , batch_size = 32 , prefetch_size = None ,
98- checkpoint_dir = None , scaling_learning_rate = True , log_step_size = 1 , validation_dataset = None ,
99- build_validation_func = None , max_iteration = float ('inf' )
97+ self , training_dataset , build_training_func , optimizer , optimizer_args , batch_size = 32 , prefetch_size = None ,
98+ checkpoint_dir = None , scaling_learning_rate = True , log_step_size = 1 , validation_dataset = None ,
99+ build_validation_func = None , max_iteration = float ('inf' )
100100 ):
101101 # Initialize Horovod.
102102 hvd .init ()
@@ -395,9 +395,9 @@ def create_task_spec_def():
395395
396396@deprecated (date = "2018-10-30" , instructions = "Using the TensorLayer distributed trainer." )
397397def create_distributed_session (
398- task_spec = None , checkpoint_dir = None , scaffold = None , hooks = None , chief_only_hooks = None , save_checkpoint_secs = 600 ,
399- save_summaries_steps = object (), save_summaries_secs = object (), config = None , stop_grace_period_secs = 120 ,
400- log_step_count_steps = 100
398+ task_spec = None , checkpoint_dir = None , scaffold = None , hooks = None , chief_only_hooks = None , save_checkpoint_secs = 600 ,
399+ save_summaries_steps = object (), save_summaries_secs = object (), config = None , stop_grace_period_secs = 120 ,
400+ log_step_count_steps = 100
401401):
402402 """Creates a distributed session.
403403
0 commit comments