5fa1a76
1
2
3
It can happen that you forgot to add some .to(self.device) statements to internal tensors of the model, which in such a test would show in an error.
