Hi, thanks for your reply but my problem is how to do the data loading. I’ve only began using pytorch recently so I’m not yet comfortable with customising the dataset. Do you know how to change the loading of batches based on the loss?