weixin_45462252的博客

当发现使用如下命令加载保存好的模型，而保存好的模型比实际搭建的模型每一层的名字都多了.moudle前缀时：checkpoint = torch.load('model.pth.tar')model.load_state_dict(checkpoint['state_dict'])可以从很多大佬的教程中找到去掉前缀的方法了，在这里就不细说了，看起来也很麻烦这里说一下为什么会出现这个.moudle前缀，搞清楚这一点可以从根源上避免这个问题这是因为我们在训练的代码中使用了“torch.nn.DataP

目录跑通POSECNN 代码IntroductionCitationInstallation1.Install TensorFlow. I usually compile the source code of tensorflow locally.2.Compile the new layers under $ROOT/lib we introduce in PoseCNN.3.下载VGG16的权...