报错的原因在于Pytorch0.4之后,在BN层后新增加了track_running_stats这个参数。 在调用预训练参数模型是,官方给定的预训练模型是在pytorch0.4之前,因此,调用预训练参数时,需要过滤掉“num_batches_tracked”。 以resnet50为例: 为了加载不同层的权重,采用两个函数,如下:load_partial_param用于加载layer1, layer2, l...