利用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小生 阅读(3495) 评论(0) 推荐(0) 编辑