jiahao000 / orl Goto Github PK
View Code? Open in Web Editor NEW[NeurIPS 2021] ORL: Unsupervised Object-Level Representation Learning from Scene Images
License: Apache License 2.0
[NeurIPS 2021] ORL: Unsupervised Object-Level Representation Learning from Scene Images
License: Apache License 2.0
hello, could you share your train2017_knn_instance_correspondence.json ?
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
...
Hello ,
I am getting an error in the first step. Plz guide me to solve this error .
File "tools/train.py", line 142, in
main()
File "tools/train.py", line 132, in main
train_model(
File "/data2/dl/grp20/ORL/openselfsup/apis/train.py", line 97, in train_model
_dist_train(
File "/data2/dl/grp20/ORL/openselfsup/apis/train.py", line 233, in _dist_train
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/mmcv/runner/epoch_based_runner.py", line 122, in run
epoch_runner(data_loaders[i], **kwargs)
File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/mmcv/runner/epoch_based_runner.py", line 29, in train
self.call_hook('before_train_iter')
File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/mmcv/runner/base_runner.py", line 282, in call_hook
getattr(hook, fn_name)(self)
File "/data2/dl/grp20/ORL/openselfsup/hooks/byol_hook.py", line 26, in before_train_iter
assert hasattr(runner.model.module, 'momentum'),
File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/torch/nn/modules/module.py", line 947, in getattr
raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'BYOL' object has no attribute 'module'
When train the first stage, I met the error "init() got an unexpected keyword argument 'prefetch'". Have you changed somgthing about the environment ”open-mmlab“?
A declarative, efficient, and flexible JavaScript library for building user interfaces.
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google ❤️ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.