RuntimeError: result type Float can‘t be cast to the desired output type __int64;
找到loss.py
将gain = torch.ones(7, device=targets.device)修改为gain = torch.ones(7, device=targets.device).long()
Ctrl+F(可能用到的快捷键)
找到loss.py
将gain = torch.ones(7, device=targets.device)修改为gain = torch.ones(7, device=targets.device).long()
Ctrl+F(可能用到的快捷键)