auto frontend

This commit is contained in:
游雁 2024-06-06 19:48:40 +08:00
parent d6fedbfa77
commit 64941d99cc

View File

@ -481,7 +481,7 @@ class Trainer:
step_in_epoch=self.step_in_epoch,
batch_num_epoch=batch_num_epoch,
lr=lr,
loss=loss.detach().cpu().item(),
loss=accum_grad * loss.detach().cpu().item(),
speed_stats=speed_stats,
stats=stats,
writer=writer,