训练时报错 TypeError: new_feature_map() missing 1 required positional argument: 'device' ，请问这个要怎么解决？ 谢谢！

[apple2333cream]

File "/home/wzping/project/linear-transformer-for-table-recognition/models.py", line 60, in forward
self.feature_map.new_feature_map()
TypeError: new_feature_map() missing 1 required positional argument: 'device'

[yang-chenyu104]

你的代码能运行吗，我的出现一些环境问题undefined symbol:出现CUDA问题不匹配的啥的