Skip to content

Commit

Permalink
Merge pull request facebookresearch#15 from eltociear/patch-1
Browse files Browse the repository at this point in the history
Update logging.py
  • Loading branch information
MidoAssran authored Feb 19, 2024
2 parents e4d7581 + f06c887 commit 851a28e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/utils/logging.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ def grad_logger(named_params):


def adamw_logger(optimizer):
""" logging magintude of first and second momentum buffers in adamw """
""" logging magnitude of first and second momentum buffers in adamw """
# TODO: assert that optimizer is instance of torch.optim.AdamW
state = optimizer.state_dict().get('state')
exp_avg_stats = AverageMeter()
Expand Down

0 comments on commit 851a28e

Please sign in to comment.