将代码中torch.save(net.state_dict(), '%s/net_%03d.pth' % (args.outf, epoch + 1))
改为torch.save(net.module, '%s/net_%03d.pth' % (args.outf, epoch + 1))
即可解决 报错'collections.OrderedDict' object has no attribute 'eval'
。源文件已经修改
比赛链接点击这里:“华为云杯”2019人工智能创新应用大赛
直接运行src/train.py
文件即可