File tree Expand file tree Collapse file tree 1 file changed +6
-7
lines changed
Expand file tree Collapse file tree 1 file changed +6
-7
lines changed Original file line number Diff line number Diff line change @@ -353,13 +353,12 @@ def main(args):
353353 assert False , f'Unknown optimizer { opt } '
354354
355355 if is_master (args ):
356- if is_master (args ):
357- defaults = copy .deepcopy (optimizer .defaults )
358- defaults ['weight_decay' ] = args .wd
359- defaults = ', ' .join ([f'{ k } : { v } ' for k , v in defaults .items ()])
360- logging .info (
361- f'Created { type (optimizer ).__name__ } ({ args .opt } ) optimizer: { defaults } '
362- )
356+ defaults = copy .deepcopy (optimizer .defaults )
357+ defaults ['weight_decay' ] = args .wd
358+ defaults = ', ' .join ([f'{ k } : { v } ' for k , v in defaults .items ()])
359+ logging .info (
360+ f'Created { type (optimizer ).__name__ } ({ args .opt } ) optimizer: { defaults } '
361+ )
363362
364363 if args .horovod :
365364 optimizer = hvd .DistributedOptimizer (optimizer , named_parameters = model .named_parameters ())
You can’t perform that action at this time.
0 commit comments