该问题通常由于原实验为分布式环境,因此在操作时产生了错误的值传递,要解决该问题只需要定位到对应的device变量或model变量处,然后做出以下更改:
#映射到cpu
model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location='cpu').to('cpu')
#映射到指定GPU
model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location=lambda storage, loc: storage.cuda()).to('cuda')
#简化版GPU映射
model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location='cuda:0').to('cuda:0')
#动态映射
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location=device).to(device)