Train model error - Githubissues

maartenbuyl / rankformer

RankFormer: Listwise Learning-to-Rank Using Listwide Labels (KDD 2023).

MIT License

17 stars 0 forks source link

Train model error #1

Open Lin-UH opened 5 months ago

Lin-UH commented 5 months ago

RuntimeError:Unknown device: 1. If you have recently updated the caffe2.proto file to add a new device type, did you forget to update the DeviceTypeName() function to reflect such recent changes?

Can anyone help with this error?

maartenbuyl commented 5 months ago

Hi!

Could you share more about the code and stacktrace that caused this error? I couldn't find much about it online.