Skip to content

Commit

Permalink
wandb step fix
Browse files Browse the repository at this point in the history
  • Loading branch information
gabrielilharco committed Aug 28, 2023
1 parent ab1ae01 commit 7007bf1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/training/train.py
Original file line number Diff line number Diff line change
@@ -225,7 +225,7 @@ def train_one_epoch(model, data, loss, epoch, optimizer, scaler, scheduler, dist
tb_writer.add_scalar(name, val, step)
if args.wandb:
assert wandb is not None, 'Please install wandb.'
wandb.log({name: val, 'step': step})
wandb.log({name: val}, step=step)

# resetting batch / data time meters per log window
batch_time_m.reset()

0 comments on commit 7007bf1

Please sign in to comment.