Do you have a repo with all the code you are using for this? (With model class, dataset class, etc). I can’t seem to find a problem in this part of the code.
Is there any random function in your model class that is not switched of when calling eval ?