Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Problem loading checkpoint in downstream object detection #2

Open
alejandrosatis opened this issue Feb 27, 2022 · 2 comments
Open

Problem loading checkpoint in downstream object detection #2

alejandrosatis opened this issue Feb 27, 2022 · 2 comments

Comments

@alejandrosatis
Copy link

Hi,

First of all I'd like to congratulate the authors for the great work and shared repo.

I have successfully followed all 3 unsupervised stages of pre-training and executed the convert-pretrain-to-detectron2.py script to perform final conversion of the model.

However, when using pretrained model for downstream task Object Detection it seems model does not get recognized (WARNING [02/27 20:12:26 d2.checkpoint.c2_model_loading]: No weights in checkpoint matched with model.) and no parameter is loaded as pretrained model. Am I missing anything? wrong configuration? Thank you in advance.

Here the log:

[02/27 20:12:26 fvcore.common.checkpoint]: [Checkpointer] Loading from /home/alejandro/satisai/py_workspace/ORL/weights/stage3_epoch_800_converted.pkl ...                                      [510/1839]
[02/27 20:12:26 fvcore.common.checkpoint]: Reading a file from 'OpenSelfSup'
WARNING [02/27 20:12:26 d2.checkpoint.c2_model_loading]: No weights in checkpoint matched with model.
WARNING [02/27 20:12:26 fvcore.common.checkpoint]: Some model parameters or buffers are not found in the checkpoint:
backbone.bottom_up.res2.0.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.0.conv1.weight
backbone.bottom_up.res2.0.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.0.conv2.weight
backbone.bottom_up.res2.0.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.0.conv3.weight
backbone.bottom_up.res2.0.shortcut.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.0.shortcut.weight
backbone.bottom_up.res2.1.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.1.conv1.weight
backbone.bottom_up.res2.1.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.1.conv2.weight
backbone.bottom_up.res2.1.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.1.conv3.weight
backbone.bottom_up.res2.2.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.2.conv1.weight
backbone.bottom_up.res2.2.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.2.conv2.weight
backbone.bottom_up.res2.2.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res2.2.conv3.weight
backbone.bottom_up.res3.0.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.0.conv1.weight
backbone.bottom_up.res3.0.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.0.conv2.weight
backbone.bottom_up.res3.0.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.0.conv3.weight
backbone.bottom_up.res3.0.shortcut.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.0.shortcut.weight
backbone.bottom_up.res3.1.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.1.conv1.weight
backbone.bottom_up.res3.1.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.1.conv2.weight
backbone.bottom_up.res3.1.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.1.conv3.weight
backbone.bottom_up.res3.2.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.2.conv1.weight
backbone.bottom_up.res3.2.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.2.conv2.weight
backbone.bottom_up.res3.2.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.2.conv3.weight
backbone.bottom_up.res3.3.conv1.norm.{bias, running_mean, running_var, weight}                                                                                                                  [466/1839]
backbone.bottom_up.res3.3.conv1.weight
backbone.bottom_up.res3.3.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.3.conv2.weight
backbone.bottom_up.res3.3.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res3.3.conv3.weight
backbone.bottom_up.res4.0.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.0.conv1.weight
backbone.bottom_up.res4.0.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.0.conv2.weight
backbone.bottom_up.res4.0.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.0.conv3.weight
backbone.bottom_up.res4.0.shortcut.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.0.shortcut.weight
backbone.bottom_up.res4.1.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.1.conv1.weight
backbone.bottom_up.res4.1.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.1.conv2.weight
backbone.bottom_up.res4.1.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.1.conv3.weight
backbone.bottom_up.res4.2.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.2.conv1.weight
backbone.bottom_up.res4.2.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.2.conv2.weight
backbone.bottom_up.res4.2.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.2.conv3.weight
backbone.bottom_up.res4.3.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.3.conv1.weight
backbone.bottom_up.res4.3.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.3.conv2.weight
backbone.bottom_up.res4.3.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.3.conv3.weight
backbone.bottom_up.res4.4.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.4.conv1.weight
backbone.bottom_up.res4.4.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.4.conv2.weight
backbone.bottom_up.res4.4.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.4.conv3.weight
backbone.bottom_up.res4.5.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.5.conv1.weight
backbone.bottom_up.res4.5.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.5.conv2.weight
backbone.bottom_up.res4.5.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res4.5.conv3.weight
backbone.bottom_up.res5.0.conv1.weight                                                                                                                                                          [421/1839]
backbone.bottom_up.res5.0.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.0.conv2.weight
backbone.bottom_up.res5.0.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.0.conv3.weight
backbone.bottom_up.res5.0.shortcut.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.0.shortcut.weight
backbone.bottom_up.res5.1.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.1.conv1.weight
backbone.bottom_up.res5.1.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.1.conv2.weight
backbone.bottom_up.res5.1.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.1.conv3.weight
backbone.bottom_up.res5.2.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.2.conv1.weight
backbone.bottom_up.res5.2.conv2.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.2.conv2.weight
backbone.bottom_up.res5.2.conv3.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.res5.2.conv3.weight
backbone.bottom_up.stem.conv1.norm.{bias, running_mean, running_var, weight}
backbone.bottom_up.stem.conv1.weight
backbone.fpn_lateral2.norm.{bias, running_mean, running_var, weight}
backbone.fpn_lateral2.weight
backbone.fpn_lateral3.norm.{bias, running_mean, running_var, weight}
backbone.fpn_lateral3.weight
backbone.fpn_lateral4.norm.{bias, running_mean, running_var, weight}
backbone.fpn_lateral4.weight
backbone.fpn_lateral5.norm.{bias, running_mean, running_var, weight}
backbone.fpn_lateral5.weight
backbone.fpn_output2.norm.{bias, running_mean, running_var, weight}
backbone.fpn_output2.weight
backbone.fpn_output3.norm.{bias, running_mean, running_var, weight}
backbone.fpn_output3.weight
backbone.fpn_output4.norm.{bias, running_mean, running_var, weight}
backbone.fpn_output4.weight
backbone.fpn_output5.norm.{bias, running_mean, running_var, weight}
backbone.fpn_output5.weight
proposal_generator.rpn_head.anchor_deltas.{bias, weight}
proposal_generator.rpn_head.conv.{bias, weight}
proposal_generator.rpn_head.objectness_logits.{bias, weight}
roi_heads.box_head.conv1.norm.{bias, running_mean, running_var, weight}
roi_heads.box_head.conv1.weight
roi_heads.box_head.conv2.norm.{bias, running_mean, running_var, weight}
roi_heads.box_head.conv2.weight
roi_heads.box_head.conv3.norm.{bias, running_mean, running_var, weight}                                                                                                                         [377/1839]
roi_heads.box_head.conv3.weight
roi_heads.box_head.conv4.norm.{bias, running_mean, running_var, weight}
roi_heads.box_head.conv4.weight
roi_heads.box_head.fc1.{bias, weight}
roi_heads.box_predictor.bbox_pred.{bias, weight}
roi_heads.box_predictor.cls_score.{bias, weight}
WARNING [02/27 20:12:26 fvcore.common.checkpoint]: The checkpoint state_dict contains keys that are not used by the model:
  stem.online_net.0.conv1.weight
  stem.online_net.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv1.weight
  online_net.0.res2.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv2.weight
  online_net.0.res2.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv3.weight
  online_net.0.res2.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.shortcut.weight
  online_net.0.res2.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.1.conv1.weight
  online_net.0.res2.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.1.conv2.weight
  online_net.0.res2.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.1.conv3.weight
  online_net.0.res2.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.2.conv1.weight
  online_net.0.res2.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.2.conv2.weight
  online_net.0.res2.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.2.conv3.weight
  online_net.0.res2.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.0.conv1.weight
  online_net.0.res3.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.0.conv2.weight
  online_net.0.res3.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.0.conv3.weight
  online_net.0.res3.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.0.shortcut.weight
  online_net.0.res3.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.1.conv1.weight
  online_net.0.res3.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.1.conv2.weight
  online_net.0.res3.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.1.conv3.weight
  online_net.0.res3.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
online_net.0.res3.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}                                                                                                 [332/1839]
  online_net.0.res3.2.conv2.weight
  online_net.0.res3.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.2.conv3.weight
  online_net.0.res3.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.3.conv1.weight
  online_net.0.res3.3.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.3.conv2.weight
  online_net.0.res3.3.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res3.3.conv3.weight
  online_net.0.res3.3.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.0.conv1.weight
  online_net.0.res4.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.0.conv2.weight
  online_net.0.res4.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.0.conv3.weight
  online_net.0.res4.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.0.shortcut.weight
  online_net.0.res4.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.1.conv1.weight
  online_net.0.res4.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.1.conv2.weight
  online_net.0.res4.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.1.conv3.weight
  online_net.0.res4.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.2.conv1.weight
  online_net.0.res4.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.2.conv2.weight
  online_net.0.res4.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.2.conv3.weight
  online_net.0.res4.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.3.conv1.weight
  online_net.0.res4.3.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.3.conv2.weight
  online_net.0.res4.3.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.3.conv3.weight
  online_net.0.res4.3.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.4.conv1.weight
  online_net.0.res4.4.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.4.conv2.weight
  online_net.0.res4.4.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.4.conv3.weight
  online_net.0.res4.4.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.5.conv1.weight
  online_net.0.res4.5.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}                                                                                                 [288/1839]
  online_net.0.res4.5.conv2.weight
  online_net.0.res4.5.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res4.5.conv3.weight
  online_net.0.res4.5.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.0.conv1.weight
  online_net.0.res5.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.0.conv2.weight
  online_net.0.res5.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.0.conv3.weight
  online_net.0.res5.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.0.shortcut.weight
  online_net.0.res5.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.1.conv1.weight
  online_net.0.res5.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.1.conv2.weight
  online_net.0.res5.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.1.conv3.weight
  online_net.0.res5.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.2.conv1.weight
  online_net.0.res5.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.2.conv2.weight
  online_net.0.res5.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res5.2.conv3.weight
  online_net.0.res5.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.online_net.1.fc0.{bias, weight}
  stem.online_net.1.bn0.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.online_net.1.fc1.weight
  stem.target_net.0.conv1.weight
  stem.target_net.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.0.conv1.weight
  target_net.0.res2.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.0.conv2.weight
  target_net.0.res2.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.0.conv3.weight
  target_net.0.res2.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.0.shortcut.weight
  target_net.0.res2.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.1.conv1.weight
  target_net.0.res2.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.1.conv2.weight
  target_net.0.res2.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.1.conv3.weight
  target_net.0.res2.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.2.conv1.weight                                                                                                                                                              [244/1839]
  target_net.0.res2.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.2.conv2.weight
  target_net.0.res2.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res2.2.conv3.weight
  target_net.0.res2.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.0.conv1.weight
  target_net.0.res3.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.0.conv2.weight
  target_net.0.res3.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.0.conv3.weight
  target_net.0.res3.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.0.shortcut.weight
  target_net.0.res3.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.1.conv1.weight
  target_net.0.res3.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.1.conv2.weight
  target_net.0.res3.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.1.conv3.weight
  target_net.0.res3.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.2.conv1.weight
  target_net.0.res3.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.2.conv2.weight
  target_net.0.res3.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.2.conv3.weight
  target_net.0.res3.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.3.conv1.weight
  target_net.0.res3.3.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.3.conv2.weight
  target_net.0.res3.3.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res3.3.conv3.weight
  target_net.0.res3.3.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.0.conv1.weight
  target_net.0.res4.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.0.conv2.weight
  target_net.0.res4.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.0.conv3.weight
  target_net.0.res4.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.0.shortcut.weight
  target_net.0.res4.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.1.conv1.weight
  target_net.0.res4.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.1.conv2.weight
  target_net.0.res4.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.1.conv3.weight                                                                                                                                                              [200/1839]
  target_net.0.res4.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.2.conv1.weight
  target_net.0.res4.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.2.conv2.weight
  target_net.0.res4.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.2.conv3.weight
  target_net.0.res4.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.3.conv1.weight
  target_net.0.res4.3.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.3.conv2.weight
  target_net.0.res4.3.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.3.conv3.weight
  target_net.0.res4.3.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.4.conv1.weight
  target_net.0.res4.4.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.4.conv2.weight
  target_net.0.res4.4.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.4.conv3.weight
  target_net.0.res4.4.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.5.conv1.weight
  target_net.0.res4.5.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.5.conv2.weight
  target_net.0.res4.5.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res4.5.conv3.weight
  target_net.0.res4.5.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.0.conv1.weight
  target_net.0.res5.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.0.conv2.weight
  target_net.0.res5.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.0.conv3.weight
  target_net.0.res5.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.0.shortcut.weight
  target_net.0.res5.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.1.conv1.weight
  target_net.0.res5.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.1.conv2.weight
  target_net.0.res5.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.1.conv3.weight
  target_net.0.res5.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.2.conv1.weight
  target_net.0.res5.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.2.conv2.weight
  target_net.0.res5.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  target_net.0.res5.2.conv3.weight                                                                                                                                                              [156/1839]
  target_net.0.res5.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.target_net.1.fc0.{bias, weight}
  stem.target_net.1.bn0.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.target_net.1.fc1.weight
  stem.backbone.conv1.weight
  stem.backbone.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.neck.fc0.{bias, weight}
  stem.neck.bn0.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.neck.fc1.weight
  stem.global_head.predictor.fc0.{bias, weight}
  stem.global_head.predictor.bn0.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.global_head.predictor.fc1.weight
  stem.local_intra_head.predictor.fc0.{bias, weight}
  stem.local_intra_head.predictor.bn0.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.local_intra_head.predictor.fc1.weight
  stem.local_inter_head.predictor.fc0.{bias, weight}
  stem.local_inter_head.predictor.bn0.{bias, num_batches_tracked, running_mean, running_var, weight}
  stem.local_inter_head.predictor.fc1.weight
  backbone.res2.0.conv1.weight
  backbone.res2.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.0.conv2.weight
  backbone.res2.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.0.conv3.weight
  backbone.res2.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.0.shortcut.weight
  backbone.res2.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.1.conv1.weight
  backbone.res2.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.1.conv2.weight
  backbone.res2.1.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.1.conv3.weight
  backbone.res2.1.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.2.conv1.weight
  backbone.res2.2.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.2.conv2.weight
  backbone.res2.2.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res2.2.conv3.weight
  backbone.res2.2.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res3.0.conv1.weight
  backbone.res3.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res3.0.conv2.weight
  backbone.res3.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  backbone.res3.0.conv3.weight
  ...
@alejandrosatis
Copy link
Author

Problem was the config, now I used this one R_50_C4_1x_imagenet.yaml and seems to be way more compatible.

[02/28 09:04:26 fvcore.common.checkpoint]: [Checkpointer] Loading from /home/alejandro/satisai/py_workspace/ORL/weights/stage3_epoch_800_converted.pkl ...
[02/28 09:04:26 fvcore.common.checkpoint]: Reading a file from 'OpenSelfSup'
[02/28 09:04:26 d2.checkpoint.c2_model_loading]: Following weights matched with submodule backbone:
| Names in Model    | Names in Checkpoint                                                                                                 | Shapes                                            |
|:------------------|:--------------------------------------------------------------------------------------------------------------------|:--------------------------------------------------|
| res2.0.conv1.*    | backbone.res2.0.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,1,1)            |
| res2.0.conv2.*    | backbone.res2.0.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,3,3)            |
| res2.0.conv3.*    | backbone.res2.0.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res2.0.shortcut.* | backbone.res2.0.shortcut.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight} | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res2.1.conv1.*    | backbone.res2.1.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,256,1,1)           |
| res2.1.conv2.*    | backbone.res2.1.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,3,3)            |
| res2.1.conv3.*    | backbone.res2.1.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res2.2.conv1.*    | backbone.res2.2.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,256,1,1)           |
| res2.2.conv2.*    | backbone.res2.2.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,3,3)            |
| res2.2.conv3.*    | backbone.res2.2.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res3.0.conv1.*    | backbone.res3.0.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,256,1,1)      |
| res3.0.conv2.*    | backbone.res3.0.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.0.conv3.*    | backbone.res3.0.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res3.0.shortcut.* | backbone.res3.0.shortcut.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight} | (512,) () (512,) (512,) (512,) (512,256,1,1)      |
| res3.1.conv1.*    | backbone.res3.1.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,512,1,1)      |
| res3.1.conv2.*    | backbone.res3.1.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.1.conv3.*    | backbone.res3.1.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res3.2.conv1.*    | backbone.res3.2.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,512,1,1)      |
| res3.2.conv2.*    | backbone.res3.2.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.2.conv3.*    | backbone.res3.2.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res3.3.conv1.*    | backbone.res3.3.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,512,1,1)      |
| res3.3.conv2.*    | backbone.res3.3.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.3.conv3.*    | backbone.res3.3.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res4.0.conv1.*    | backbone.res4.0.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,512,1,1)      |
| res4.0.conv2.*    | backbone.res4.0.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.0.conv3.*    | backbone.res4.0.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.0.shortcut.* | backbone.res4.0.shortcut.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight} | (1024,) () (1024,) (1024,) (1024,) (1024,512,1,1) |
| res4.1.conv1.*    | backbone.res4.1.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.1.conv2.*    | backbone.res4.1.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.1.conv3.*    | backbone.res4.1.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.2.conv1.*    | backbone.res4.2.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.2.conv2.*    | backbone.res4.2.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.2.conv3.*    | backbone.res4.2.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.3.conv1.*    | backbone.res4.3.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.3.conv2.*    | backbone.res4.3.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.3.conv3.*    | backbone.res4.3.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.4.conv1.*    | backbone.res4.4.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.4.conv2.*    | backbone.res4.4.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.4.conv3.*    | backbone.res4.4.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.5.conv1.*    | backbone.res4.5.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.5.conv2.*    | backbone.res4.5.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.5.conv3.*    | backbone.res4.5.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
WARNING [02/28 09:04:26 fvcore.common.checkpoint]: Some model parameters or buffers are not found in the checkpoint:
backbone.stem.conv1.norm.{bias, running_mean, running_var, weight}
backbone.stem.conv1.weight
proposal_generator.rpn_head.anchor_deltas.{bias, weight}
proposal_generator.rpn_head.conv.{bias, weight}
proposal_generator.rpn_head.objectness_logits.{bias, weight}
roi_heads.box_predictor.bbox_pred.{bias, weight}
roi_heads.box_predictor.cls_score.{bias, weight}
roi_heads.res5.0.conv1.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.conv1.weight
roi_heads.res5.0.conv2.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.conv2.weight
roi_heads.res5.0.conv3.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.conv3.weight
roi_heads.res5.0.shortcut.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.shortcut.weight
roi_heads.res5.1.conv1.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.1.conv1.weight
roi_heads.res5.1.conv2.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.1.conv2.weight
roi_heads.res5.1.conv3.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.1.conv3.weight
roi_heads.res5.2.conv1.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.2.conv1.weight
roi_heads.res5.2.conv2.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.2.conv2.weight
roi_heads.res5.2.conv3.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.2.conv3.weight
roi_heads.res5.norm.{bias, running_mean, running_var, weight}
WARNING [02/28 09:04:26 fvcore.common.checkpoint]: The checkpoint state_dict contains keys that are not used by the model:
  online_net.0.res2.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv1.weight
  online_net.0.res2.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv2.weight
  online_net.0.res2.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv3.weight
  online_net.0.res2.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.shortcut.weight
  online_net.0.res2.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
... 

@592McAvoy
Copy link

Problem was the config, now I used this one R_50_C4_1x_imagenet.yaml and seems to be way more compatible.

[02/28 09:04:26 fvcore.common.checkpoint]: [Checkpointer] Loading from /home/alejandro/satisai/py_workspace/ORL/weights/stage3_epoch_800_converted.pkl ...
[02/28 09:04:26 fvcore.common.checkpoint]: Reading a file from 'OpenSelfSup'
[02/28 09:04:26 d2.checkpoint.c2_model_loading]: Following weights matched with submodule backbone:
| Names in Model    | Names in Checkpoint                                                                                                 | Shapes                                            |
|:------------------|:--------------------------------------------------------------------------------------------------------------------|:--------------------------------------------------|
| res2.0.conv1.*    | backbone.res2.0.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,1,1)            |
| res2.0.conv2.*    | backbone.res2.0.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,3,3)            |
| res2.0.conv3.*    | backbone.res2.0.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res2.0.shortcut.* | backbone.res2.0.shortcut.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight} | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res2.1.conv1.*    | backbone.res2.1.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,256,1,1)           |
| res2.1.conv2.*    | backbone.res2.1.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,3,3)            |
| res2.1.conv3.*    | backbone.res2.1.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res2.2.conv1.*    | backbone.res2.2.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,256,1,1)           |
| res2.2.conv2.*    | backbone.res2.2.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (64,) () (64,) (64,) (64,) (64,64,3,3)            |
| res2.2.conv3.*    | backbone.res2.2.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,64,1,1)       |
| res3.0.conv1.*    | backbone.res3.0.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,256,1,1)      |
| res3.0.conv2.*    | backbone.res3.0.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.0.conv3.*    | backbone.res3.0.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res3.0.shortcut.* | backbone.res3.0.shortcut.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight} | (512,) () (512,) (512,) (512,) (512,256,1,1)      |
| res3.1.conv1.*    | backbone.res3.1.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,512,1,1)      |
| res3.1.conv2.*    | backbone.res3.1.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.1.conv3.*    | backbone.res3.1.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res3.2.conv1.*    | backbone.res3.2.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,512,1,1)      |
| res3.2.conv2.*    | backbone.res3.2.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.2.conv3.*    | backbone.res3.2.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res3.3.conv1.*    | backbone.res3.3.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,512,1,1)      |
| res3.3.conv2.*    | backbone.res3.3.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (128,) () (128,) (128,) (128,) (128,128,3,3)      |
| res3.3.conv3.*    | backbone.res3.3.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (512,) () (512,) (512,) (512,) (512,128,1,1)      |
| res4.0.conv1.*    | backbone.res4.0.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,512,1,1)      |
| res4.0.conv2.*    | backbone.res4.0.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.0.conv3.*    | backbone.res4.0.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.0.shortcut.* | backbone.res4.0.shortcut.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight} | (1024,) () (1024,) (1024,) (1024,) (1024,512,1,1) |
| res4.1.conv1.*    | backbone.res4.1.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.1.conv2.*    | backbone.res4.1.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.1.conv3.*    | backbone.res4.1.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.2.conv1.*    | backbone.res4.2.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.2.conv2.*    | backbone.res4.2.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.2.conv3.*    | backbone.res4.2.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.3.conv1.*    | backbone.res4.3.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.3.conv2.*    | backbone.res4.3.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.3.conv3.*    | backbone.res4.3.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.4.conv1.*    | backbone.res4.4.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.4.conv2.*    | backbone.res4.4.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.4.conv3.*    | backbone.res4.4.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
| res4.5.conv1.*    | backbone.res4.5.conv1.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,1024,1,1)     |
| res4.5.conv2.*    | backbone.res4.5.conv2.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (256,) () (256,) (256,) (256,) (256,256,3,3)      |
| res4.5.conv3.*    | backbone.res4.5.conv3.{norm.bias,norm.num_batches_tracked,norm.running_mean,norm.running_var,norm.weight,weight}    | (1024,) () (1024,) (1024,) (1024,) (1024,256,1,1) |
WARNING [02/28 09:04:26 fvcore.common.checkpoint]: Some model parameters or buffers are not found in the checkpoint:
backbone.stem.conv1.norm.{bias, running_mean, running_var, weight}
backbone.stem.conv1.weight
proposal_generator.rpn_head.anchor_deltas.{bias, weight}
proposal_generator.rpn_head.conv.{bias, weight}
proposal_generator.rpn_head.objectness_logits.{bias, weight}
roi_heads.box_predictor.bbox_pred.{bias, weight}
roi_heads.box_predictor.cls_score.{bias, weight}
roi_heads.res5.0.conv1.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.conv1.weight
roi_heads.res5.0.conv2.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.conv2.weight
roi_heads.res5.0.conv3.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.conv3.weight
roi_heads.res5.0.shortcut.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.0.shortcut.weight
roi_heads.res5.1.conv1.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.1.conv1.weight
roi_heads.res5.1.conv2.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.1.conv2.weight
roi_heads.res5.1.conv3.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.1.conv3.weight
roi_heads.res5.2.conv1.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.2.conv1.weight
roi_heads.res5.2.conv2.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.2.conv2.weight
roi_heads.res5.2.conv3.norm.{bias, running_mean, running_var, weight}
roi_heads.res5.2.conv3.weight
roi_heads.res5.norm.{bias, running_mean, running_var, weight}
WARNING [02/28 09:04:26 fvcore.common.checkpoint]: The checkpoint state_dict contains keys that are not used by the model:
  online_net.0.res2.0.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv1.weight
  online_net.0.res2.0.conv2.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv2.weight
  online_net.0.res2.0.conv3.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.conv3.weight
  online_net.0.res2.0.shortcut.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
  online_net.0.res2.0.shortcut.weight
  online_net.0.res2.1.conv1.norm.{bias, num_batches_tracked, running_mean, running_var, weight}
... 

I met the same problem. It seems that the converted state_dict key names do not meet the detectron2 requirments.

Add k = k.replace("online_net.0.", "") in the for loop in convert-pretrain-to-detectron2.py can fix the problem.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants