2021年6月24日
多卡训练的state_dict
摘要： 在pytorch多卡训练得到的pth.tar形式的checkpoint中一个问题是key都被加上了module,例如原来的'fc.weight'就变成了'module.fc.weight',在加载进单卡下的模型时如果命令为model.load_state_dict(state_dict, stric 阅读全文
posted @ 2021-06-24 15:17 别再闹了 阅读(279) 评论(0) 推荐(0) 编辑