0.4.1训练的模型会有 num_batches_tracked结尾的参数,事实上这些参数记录的信息 一般情况模型并不需要。
0.4.0加载模型的时候 不会识别num_batches_tracked结尾的名字
故需要将0.4.1的模型参数中num_batches_tracked结尾的参数删除
例如价值0.4.1模型得到一个state_dict
state_dict = torch.load(model_path)['net_dict']
通过以下语句过滤下
state_dict = {k: v for k, v in state_dict.items() if 'num_batches_tracked' not in k}
然后就行了。
哈哈 我好强。