DatatransformsMagNet2 from data import MagNetChallange2 data=MagNetChallange2("78") data_train_loader = DataLoader(data, batch_size=64, shuffle=True)