@JuanFMontesinos
Should I call transformations in __init__?
When I load data, I called transformations like under.
Let me know where should I edit the code 
Thank you.
# load train data
transform_train = transforms.Compose([
    transforms.RandomHorizontalFlip(),
    transforms.ToTensor(),
    transforms.Normalize(mean=(0.4914, 0.4824, 0.4467),
                             std=(0.2471, 0.2436, 0.2616))
])

train_dataset = datasets.CIFAR10(root='../../data/cifar10/',
                                 train=True,
                                 transform=transform_train,
                                 download=True)