torch
torchvision
torchvision.transforms as transforms
torch.utils.data import DataLoader
numpy as np
scipy import stats