I think I found out the problem here. According to this topic ModuleNotFoundError: No module named 'network' - #3 by Oscar_Rangel concerning the saving of full models, ptrblck said that :
If you are using this approach:
model = torch.load(path), you would need to make sure that all necessary files are in the corresponding folders as they were while storing the model.
It looks like it does apply here too when I use the same organisation as the place where I saved the dataLoader, when I saved it.