search

reczoo / RecZoo

A curated model zoo for recommendation tasks
Apache License 2.0
170 stars 43 forks source link

device问题 #39

Closed fgriver closed 5 months ago

fgriver commented 5 months ago

在运行作者代码的时候经常会碰到RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)，需要多次运行以调整device问题，请问作者在上传代码的时候可以正常运行吗？

zhujiem commented 5 months ago

大概率是torch版本问题。建议最好follow README中的版本跑代码。测试过程中没有上述问题。