pykt-team / pykt-toolkit

pyKT: A Python Library to Benchmark Deep Learning based Knowledge Tracing Models
https://pykt.org
MIT License
194 stars 53 forks source link

关于运行多个模型都出现‘RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! ’的报错 #158

Closed xiangxin-oss closed 6 months ago

xiangxin-oss commented 6 months ago

您好,我在用assist2015数据集跑akt、atdkt、simplekt和sparsekt等模型时都报了下述图片上的错误,求助如何

需要在具体代码后加to(device)。