将loss.py中的
gain = torch.ones(7, device=targets.device)
改为
gain = torch.ones(7, device=targets.device).long()
原因是新的新版本的torch无法自动执行此转换,旧版本torch可以。
将loss.py中的
gain = torch.ones(7, device=targets.device)
改为
gain = torch.ones(7, device=targets.device).long()
原因是新的新版本的torch无法自动执行此转换,旧版本torch可以。