Я использую следующий код для увеличения данных MNIST:
train_loader = torch.utils.data.DataLoader(
datasets.MNIST('../data', train=True, download=True,
transform=transforms.Compose([
transforms.RandomHorizontalFlip(),
transforms.RandomResizedCrop(28),
transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,)),
])),
batch_size=args.batch_size, shuffle=True, **kwargs)
У меня есть вопрос по поводу строки transforms.Normalize((0.1307,), (0.3081,))
, 0,1307 и 0,3801 - среднее значение и стандартное отклонение исходного набора данных MNIST. Они должны были быть изменены после этого увеличения. Так должен ли я использовать новое среднее значение и отклонение для нормализации? Еще один вопрос: я должен сделать то же самое увеличение на тестовом наборе? Если нет, то тренировки с дополнением и тестированием будут проходить из другого дистрибутива, верно?