2023年8月15日
模型超参数基本都没改，测试时加载模型报模型结构不匹配，设置模糊加载模型即：model.load_state_dict(torch.load(model_path), strict=Fasle)，但效果出奇的差
摘要： # 原因 多卡训练；单卡模糊加载进行测试。 训练时，通过torch.nn.DataParallel(self.model)进行多卡并行训练；测试时，用单卡模糊加载保存的模型权重，很多模型参数都没有加载成功，自然会导致测试效果很差。 # 解决方法 测试时，使用多卡加载模型时，删掉'module.'前缀 阅读全文
posted @ 2023-08-15 17:28 Kurie 阅读(121) 评论(0) 推荐(0) 编辑