I think that I’ve figured it out.
Using this:
totalLoss += test_module.loss
Gives me the “rounded” result.
While using:
totalLoss += test_module.loss.item()
Or:
totalLoss += float(test_module.loss)
Gives me the correct result.
But I can only use .item() or float() for printing the values. I don’t know how to add my loss variables together in a way that doesn’t result in this “rounding”.