fa961f2290c3144f1c5a5c9b8a484610ab835032,tests/defences/test_pixel_defend.py,TestPixelDefend,test_one_channel,#TestPixelDefend#,58

Before Change

    def test_one_channel(self):
        (x_train, _), (_, _), _, _ = load_mnist()
        x_train = x_train[:2]

        x_train = x_train[:, 10:15, 15:20, :]
        preprocess = PixelDefend(eps=5, pixel_cnn=self.pixelcnn)
        defended_x, _ = preprocess(x_train)

After Change

        x_train = x_train[:2, 10:15, 15:20, :]

        // Define the network
        model = ModelImage()
        loss_fn = nn.CrossEntropyLoss()
        optimizer = optim.Adam(model.parameters(), lr=0.01)
        self.pixelcnn = PyTorchClassifier(model=model, loss=loss_fn, optimizer=optimizer, input_shape=(1, 28, 28),
                                          nb_classes=10, clip_values=(0, 1))
        preprocess = PixelDefend(eps=5, pixel_cnn=self.pixelcnn)
        defended_x, _ = preprocess(x_train)

Instances