jungscott

assert outputs.is_cuda or outputs.device.type == 'xla'

5

File ~\anaconda3\envs\pvt2\lib\site-packages\torch\cuda\amp\grad_scaler.py:161 in scale assert outputs.is_cuda or outputs.device.type == 'xla' AssertionError