Directly comparing floating point values might fail due to the limited floating point precision, which would be visible if e.g. a non-deterministic algorithm is used so use torch.allclose instead.



 aniketmaurya:

The prediction final results are correct mostly but there is a significant change in raw logit values.


How large is this significant change if you print the abs().max() of the difference?