pytorch0.4.0版本如何加载torch0.4.1模型,看这里

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}

然后就行了。

哈哈 我好强。

你可能感兴趣的:(pytorch0.4.0版本如何加载torch0.4.1模型,看这里)