随笔分类 - pytorch
摘要：# 原因 多卡训练；单卡模糊加载进行测试。 训练时，通过torch.nn.DataParallel(self.model)进行多卡并行训练；测试时，用单卡模糊加载保存的模型权重，很多模型参数都没有加载成功，自然会导致测试效果很差。 # 解决方法 测试时，使用多卡加载模型时，删掉'module.'前缀
阅读全文
摘要：I 验证网络结构是否有误 初始化一个符合网络的输入数据 input = torch.ones((64, 3, 32, 32)) 将输入数据传进网络，看是否报错 print(network(input).shape) II 修改已知网络(比如vgg16) vgg16_false = torchvisi
阅读全文
