Creating loaders for training and validation