Skip to content

Commit 004aa0c

Browse files
shreyaskamathkmrwightman
authored andcommitted
Refactor optimizer logging in main function for clarity and conciseness
1 parent a87f11e commit 004aa0c

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

src/open_clip_train/main.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff 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())

0 commit comments

Comments
 (0)