def _get_cutout_train_transform(self):
     transform = torchvision.transforms.Compose([
         transforms.normalize(self.mean, self.std),
         transforms.cutout(self.config['cutout_size'],
                           self.config['cutout_prob'],
                           self.config['cutout_inside']),
         transforms.to_tensor(),
     ])
     return transform
 def _get_cutout_train_transform(self):
     transform = torchvision.transforms.Compose([
         torchvision.transforms.RandomCrop(28, padding=4),
         torchvision.transforms.RandomHorizontalFlip(),
         transforms.normalize(self.mean, self.std),
         transforms.cutout(self.config['cutout_size'],
                           self.config['cutout_prob'],
                           self.config['cutout_inside']),
         transforms.to_tensor(),
     ])
     return transform
 def _add_cutout(self):
     transform = transforms.cutout(self.config['cutout_size'],
                                   self.config['cutout_prob'],
                                   self.config['cutout_inside'])
     self._train_transforms.append(transform)