jonasrauber / foolbox-native Goto Github PK

AssertionError                            Traceback (most recent call last)
<ipython-input-94-2fdb7e703e80> in <module>
----> 1 fbn.utils.accuracy(fmodel, images, labels)

***/lib/python3.6/site-packages/foolbox/ext/native/utils.py in accuracy(fmodel, inputs, labels)
      8 
      9 def accuracy(fmodel, inputs, labels):
---> 10     logits = ep.astensor(fmodel.forward(inputs))
     11     predictions = logits.argmax(axis=-1)
     12     accuracy = (predictions == labels).float32().mean()

***/lib/python3.6/site-packages/foolbox/ext/native/models/pytorch.py in forward(self, inputs)
     74     def forward(self, inputs):
     75         x = inputs
---> 76         assert x.device == self.device
     77         x = self._preprocess(x)
     78         x = self._model(x)

AssertionError: