> # choose the training and test datasets
> train_data = datasets.ImageFolder(data+"/train", transform=transform_train)
> test_data = datasets.ImageFolder(data+"/val", transform = transform_test)
> 
> dataloader_train = torch.utils.data.DataLoader(train_data, batch_size=32, shuffle=True, num_workers=0)
> dataloader_test = torch.utils.data.DataLoader(test_data, batch_size=32, num_workers=0)

but just running and then vanish.