vltanh / torchan

General template for my PyTorch projects.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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).