spectre-defense icon indicating copy to clipboard operation
spectre-defense copied to clipboard
verified publisher icon SewoongLab

Fix batch loss normalization in mini_train

Open jhayase opened this issue 4 years ago • 0 comments

The normalization of the loss in mini_train is slightly off if the batch does not have size batch_size, which can be the case when drop_last is false.

jhayase avatar Oct 29 '21 18:10 jhayase