blog
blog copied to clipboard
Update pytorch-ddp-accelerate-transformers.md
As the model is on 'cuda', the test data and labels also should be on the same device for evaluation. I guess this is just a copy paste issue