Okay - I’m in the processing of creating some new environments so will update you in a few. Could you  lastly share the full call to the get_dataloaders function? It seems you are not using the val_split command which is the whole point of me using SubsetRandomSampler.