报错信息 报错原因 开始遇到这个问题是真的有点懵,因为这是到了最后一个test.py的文件了,但是确不能跑通。 查了很多博客,终于站在巨人的肩膀上,找到了原因!!! 原来的代码模型是使用的多个GPU进行训练的,而当我实现时,只使用了一个,所以报错。 原本代码:torch.load(self.cfg.snapshot) 修改代码:torch.load(self.cfg.snapshot,map_location = 'cpu' )