Metrics update
ShkarupaDC opened this issue · 0 comments
ShkarupaDC commented
I think there is a small bug here:
losses.update(loss.item(), data_loader.batch_size)
because last batch can be smaller, than 'batch size' defined in DataLoader. So metrics are going to be computed with some error