ContraRec训练报错

[Donowert]

作者您好，我用pytorch1.8版本测试ContraRec的训练发生了下图的错误。 请问是因为torch不是1.1版本导致的吗？

[THUwangcy]

应该是的，可能需要把 torch.arange(batch_size) 规定为 torch.long 类型，默认应该是 float

[Donowert]

谢谢大佬，我试试。

[Donowert]

我加了一个long类型，还是报错

[THUwangcy]

lengths 也加一个 .long() 试试

[Donowert]

加了以后能跑起来，但是loss一致不变

[Donowert]

这是跑完后的结果

[THUwangcy]

感觉是 torch 版本还带来了其它问题，在 1.1 下刚刚测试还是正常的，之后我再细查一下

[Donowert]

Caser我试了下好像没问题，麻烦大佬了！

[Donowert]

Caser好像也有问题，应该是版本的问题