Skip to content

Commit

Permalink
Update logging.py
Browse files Browse the repository at this point in the history
magintude -> magnitude
  • Loading branch information
eltociear authored Feb 19, 2024
1 parent 61d1e39 commit f06c887
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 f06c887

Please sign in to comment.