fc82df0ff84975f8102a291363abbc01a3986547,examples/adversarial_training_FBF.py,,,#,148

Before Change

tensor_y = torch.Tensor(y_train)

my_dataset = TensorDataset(tensor_x, tensor_y)  // create your datset
my_dataloader = DataLoader(my_dataset)  // create your dataloader

transform = transforms.Compose(
    [transforms.RandomCrop(32, padding=4), transforms.RandomHorizontalFlip(), transforms.ToTensor()]

After Change

classifier = PyTorchClassifier(
    model=model,
    clip_values=(0.0, 1.0),
    preprocessing=(cifar_mu, cifar_std),
    loss=criterion,
    optimizer=opt,

Instances