Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
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.