resnextでエラー

[kminoda]

Traceback (most recent call last):
  File "train.py", line 205, in <module>
    model = Model(config['model']).to(device)
  File "/home/minoda/git/GlobalWheatDetection/model.py", line 121, in __init__
    self.model = model_list[config['name']](**config['config'])
  File "/home/minoda/git/GlobalWheatDetection/model.py", line 108, in fasterrcnn_model
    backbone = resnet_fpn_backbone(backbone, pretrained=pretrained)
  File "/home/minoda/venv/pytorch140-py36/lib/python3.6/site-packages/torchvision/models/detection/backbone_utils.py", line 47, in resnet_fpn_backbone
    norm_layer=misc_nn_ops.FrozenBatchNorm2d)
  File "/home/minoda/venv/pytorch140-py36/lib/python3.6/site-packages/torchvision/models/resnet.py", line 299, in resnext50_32x4d
    pretrained, progress, **kwargs)
  File "/home/minoda/venv/pytorch140-py36/lib/python3.6/site-packages/torchvision/models/resnet.py", line 224, in _resnet
    model.load_state_dict(state_dict)
  File "/home/minoda/venv/pytorch140-py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 830, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for ResNet:
        Unexpected key(s) in state_dict: "bn1.num_batches_tracked", "layer1.0.bn1.num_batches_tracked", "layer1.0.bn2.num_batches_tracked", "layer1.0.bn3.num_batches_tracked", "layer1.0.downsample.1.num_batches_tracked", "layer1.1.bn1.num_batches_tracked", "layer1.1.bn2.num_batches_tracked", "layer1.1.bn3.num_batches_tracked", "layer1.2.bn1.num_batches_tracked", "layer1.2.bn2.num_batches_tracked", "layer1.2.bn3.num_batches_tracked", "layer2.0.bn1.num_batches_tracked", "layer2.0.bn2.num_batches_tracked", "layer2.0.bn3.num_batches_tracked", "layer2.0.downsample.1.num_batches_tracked", "layer2.1.bn1.num_batches_tracked", "layer2.1.bn2.num_batches_tracked", "layer2.1.bn3.num_batches_tracked", "layer2.2.bn1.num_batches_tracked", "layer2.2.bn2.num_batches_tracked", "layer2.2.bn3.num_batches_tracked", "layer2.3.bn1.num_batches_tracked", "layer2.3.bn2.num_batches_tracked", "layer2.3.bn3.num_batches_tracked", "layer3.0.bn1.num_batches_tracked", "layer3.0.bn2.num_batches_tracked", "layer3.0.bn3.num_batches_tracked", "layer3.0.downsample.1.num_batches_tracked", "layer3.1.bn1.num_batches_tracked", "layer3.1.bn2.num_batches_tracked", "layer3.1.bn3.num_batches_tracked", "layer3.2.bn1.num_batches_tracked", "layer3.2.bn2.num_batches_tracked", "layer3.2.bn3.num_batches_tracked", "layer3.3.bn1.num_batches_tracked", "layer3.3.bn2.num_batches_tracked", "layer3.3.bn3.num_batches_tracked", "layer3.4.bn1.num_batches_tracked", "layer3.4.bn2.num_batches_tracked", "layer3.4.bn3.num_batches_tracked", "layer3.5.bn1.num_batches_tracked", "layer3.5.bn2.num_batches_tracked", "layer3.5.bn3.num_batches_tracked", "layer4.0.bn1.num_batches_tracked", "layer4.0.bn2.num_batches_tracked", "layer4.0.bn3.num_batches_tracked", "layer4.0.downsample.1.num_batches_tracked", "layer4.1.bn1.num_batches_tracked", "layer4.1.bn2.num_batches_tracked", "layer4.1.bn3.num_batches_tracked", "layer4.2.bn1.num_batches_tracked", "layer4.2.bn2.num_batches_tracked", "layer4.2.bn3.num_batches_tracked".

[kminoda]

resnextをbackboneに指定すると出る

[kminoda]

あ、pretrained=falseにしたら治った

[Toma0916]

むむむ れずねくすとはぷれとれいんでうごかないのか、もしかしたらモデルの構造中でハードコードしてるところ直せば動くかもかも

[kminoda]

今走らせてしまったので学習終わったら確認してみようかな

[Toma0916]

これって確認した？

[kminoda]

あ 記憶から抜け落ちてますね