test_loss not normalized #131
Description
test_loss computation L459 is not scaled correctly.
Ideally, it should be multiplied by the batch_size in this line, and then divided by the size of the dataset at the end.
{{ message }}
test_loss computation L459 is not scaled correctly.
Ideally, it should be multiplied by the batch_size in this line, and then divided by the size of the dataset at the end.