直接替换即可运行,已经加了转码
ctrl F 输入 open
找到with open(opt.data') as f: 替换为以下代码
with open(opt.data,errors='ignore',encoding='utf-8') as f:
大概有两处 有的都替换了
第二 Loss.py 中存在GPU和CPU同时调用错误
indices should be either on cpu or on the same device as the indexed tensor (cpu)
ctrl F 输入 matching_matrix 找到以下代码
matching_matrix = torch.zeros_like(cost)
替换为
matching_matrix = torch.zeros_like(cost, device="cpu")
即可解决train的报错