search

OceanPang / Libra_R-CNN

Code for CVPR 2019 paper "Libra R-CNN: Towards Balanced Learning for Object Detection"
Apache License 2.0
367 stars 49 forks source link

Initialize the weight problem? #4

Closed hdjsjyl closed 5 years ago

hdjsjyl commented 5 years ago

My operation system is Ubuntu16.04, CUDA9.0, PyTorch1.1 stable, GCC5.4. When initializing the weights of resnet, it has the following problem:

issing keys in source state_dict: layer3.2.bn1.num_batches_tracked, layer4.2.bn2.num_batches_tracked, layer3.3.bn2.num_batches_tracked, layer1.2.bn3.num_batches_tracked, layer3.2.bn2.num_batches_tracked, layer4.2.bn3.num_batches_tracked, bn1.num_batches_tracked, layer1.0.bn3.num_batches_tracked, layer1.2.bn1.num_batches_tracked, layer2.0.bn2.num_batches_tracked, layer2.3.bn3.num_batches_tracked, layer2.1.bn2.num_batches_tracked, layer3.4.bn3.num_batches_tracked, layer3.5.bn3.num_batches_tracked, layer3.3.bn1.num_batches_tracked, layer1.1.bn2.num_batches_tracked, layer3.2.bn3.num_batches_tracked, layer2.1.bn1.num_batches_tracked, layer3.4.bn1.num_batches_tracked, layer2.2.bn2.num_batches_tracked, layer3.5.bn2.num_batches_tracked, layer1.0.bn1.num_batches_tracked, layer1.1.bn1.num_batches_tracked, layer2.0.bn3.num_batches_tracked, layer1.0.downsample.1.num_batches_tracked, layer2.1.bn3.num_batches_tracked, layer4.0.bn1.num_batches_tracked, layer2.3.bn1.num_batches_tracked, layer2.0.downsample.1.num_batches_tracked, layer3.0.bn3.num_batches_tracked, layer3.1.bn3.num_batches_tracked, layer3.5.bn1.num_batches_tracked, layer2.3.bn2.num_batches_tracked, layer4.1.bn2.num_batches_tracked, layer4.0.downsample.1.num_batches_tracked, layer2.0.bn1.num_batches_tracked, layer3.1.bn2.num_batches_tracked, layer3.0.downsample.1.num_batches_tracked, layer3.1.bn1.num_batches_tracked, layer3.4.bn2.num_batches_tracked, layer3.0.bn1.num_batches_tracked, layer3.3.bn3.num_batches_tracked, layer4.2.bn1.num_batches_tracked, layer1.0.bn2.num_batches_tracked, layer4.0.bn3.num_batches_tracked, layer4.1.bn1.num_batches_tracked, layer4.1.bn3.num_batches_tracked, layer2.2.bn3.num_batches_tracked, layer1.2.bn2.num_batches_tracked, layer2.2.bn1.num_batches_tracked, layer3.0.bn2.num_batches_tracked, layer1.1.bn3.num_batches_tracked, layer4.0.bn2.num_batches_tracked

OceanPang commented 5 years ago

That's right.