train_dataset = MRDataset('./data/', args.task, args.plane, transform=augmentor, train=True) validation_dataset = MRDataset('./data/', args.task, args.plane, train=False)