<reponame>TTitcombe/LotteryTicketHypothesis
import torchvision.transforms as transforms

mnist_train_transform = transforms.Compose(
    [transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))]
)

import torchvision.transforms as transforms

mnist_train_transform = transforms.Compose(
    [transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))]
)