Key Error in pretrained model

[Fly-dream12]

Thanks for your project!
I have downloaded the pretrained model diou.pth , but when I use this it runs into an error like this:

File "DIoU-pytorch-detectron-master/lib/modeling/model_builder.py", line 128, in _init_modules
resnet_utils.load_pretrained_imagenet_weights(self)
File "DIoU-pytorch-detectron-master/lib/utils/resnet_weights_helper.py", line 38, in load_pretrained_imagenet_weights
bn_mean = pretrianed_state_dict[pretrianed_name + '.running_mean']
KeyError: 'res1.bn1.running_mean'

I wonder if i have not set the right defualt settings

[Zzh-tju]

Have you try other weights？Or maybe environment problem.

[Fly-dream12]

I have tried to set LOAD_IMAGENET_PRETRAINED_WEIGHTS to False and it can succesfully train.
Is it OK?