来源:https://stackoverflow.com/questions/66091226/runtimeerror-expected-all-tensors-to-be-on-the-same-device-but-found-at-least
这是一个 pytorch 中常见的错误,表示训练或者预测过程中,有些张量不在同一个设备上
常见的原因是,模型的输入在 GPU 上,但模型不在 GPU 上
加上下面这个代码通常就能 work:
model = model.to('cuda')