将loss.py中gain = torch.ones(7, device=targets.device)改为gain = torch.ones(7, device=targets.device).long()即可。
原因是新版本的torch无法自动执行此转换,旧版本torch可以。
![](https://i-blog.csdnimg.cn/blog_migrate/79b647486fe91d8cd05fa61093723750.png)
将loss.py中gain = torch.ones(7, device=targets.device)改为gain = torch.ones(7, device=targets.device).long()即可。
原因是新版本的torch无法自动执行此转换,旧版本torch可以。