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
orl's People
Contributors
Stargazers
Watchers
orl's Issues
train2017_knn_instance_correspondence
hello, could you share your train2017_knn_instance_correspondence.json ?
Problem loading checkpoint in downstream object detection
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
...
BYOL object has no attribute module
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'
__init__() got an unexpected keyword argument 'prefetch'
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“?
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.