将代码中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人工智能创新应用大赛

比赛链接点击这里:“华为云杯”2019人工智能创新应用大赛

直接运行src/train.py文件即可

加注意力机制:1.pytorch中加入注意力机制(CBMA)。2.keras, TensorFlow中加入注意力机制

知乎详解

知乎详解点击这里

此代码为线下运行代码,不支持华为云端运行