利用resnet预训练权重，出现“bn1.num_batches_tracked”或者“layer.0.bn1.num_batches_tracked" 的解决办法
摘要：
报错的原因在于Pytorch0.4之后，在BN层后新增加了track_running_stats这个参数。 在调用预训练参数模型是，官方给定的预训练模型是在pytorch0.4之前，因此，调用预训练参数时，需要过滤掉“num_batches_tracked”。 以resnet50为例： 为了加载不同 阅读全文
posted @ 2020-08-20 19:31 RS小生 阅读(3433) 评论(0) 推荐(0) 编辑