Loss update wrongly for variable batch_size
vltanh opened this issue · comments
Last batch might be of different sizes so loss_meter.add(loss) is not correct, should be loss_meter.add(loss*bs, bs).
General template for my PyTorch projects.
vltanh opened this issue · comments
Last batch might be of different sizes so loss_meter.add(loss) is not correct, should be loss_meter.add(loss*bs, bs).