3bf391c2315bcec722961a2f4a093d1c516dbb10,art/classifiers/pytorch.py,PyTorchClassifier,__init__,#PyTorchClassifier#Any#Any#Any#Any#Any#Any#Any#Any#Any#,14

Before Change

        // Use GPU if possible
        import torch
        self._device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
        self._model.to(self._device)

    def predict(self, x, logits=False, batch_size=128):

After Change

        // Use GPU if possible
        import torch
        self._device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
        self._model.to(self._device)

    def predict(self, x, logits=False, batch_size=128):
        

Instances