ssalawu

RuntimeError: arguments are located on different GPUs

2

I get the above error when performing fine-tuning using fast-bert. I'm environment is Win10 (2 gpu), Python 3.7, Fast-Bert 1.6.5, cuda 10.1. The code I'm using is as below: `#...