在测试时报错内容:
解决:定位到221行,为
ground = intens.gather(1, actor2.view(-1, 1))
需要把actor2改成int64类型即可
在改行代码前加上:
actor2=actor2.to(torch.int64)
在测试时报错内容:
解决:定位到221行,为
ground = intens.gather(1, actor2.view(-1, 1))
需要把actor2改成int64类型即可
在改行代码前加上:
actor2=actor2.to(torch.int64)