Missing Keys and Unexpected Keys when trying to load PP_YOLOE_X #2078
Description
💡 Your Question
Hi, when I try to load the pretrained PP_YOLOE_X model, which I'm trying to fine tune, in the following way, I get the Traceback below. I saved ppyoloe_x_coco.pth into my local directory after I downloaded it from https://sg-hub-nv.s3.amazonaws.com/models/ppyoloe_x_coco.pth .
How do I get the model to load properly with the number of classes I want?
from super_gradients.training import models
from super_gradients.common.object_names import Models
model = models.get(
model_name=Models.PP_YOLOE_X,
checkpoint_path='ppyoloe_x_coco.pth',
num_classes=20) ## also tried checkpoint_num_classes=20 with same result
Traceback (most recent call last):
File "train.py", line 16, in <module>
model = models.get(
File "super-gradients/src/super_gradients/common/decorators/factory_decorator.py", line 36, in wrapper
return func(*args, **kwargs)
File "super-gradients/src/super_gradients/training/models/model_factory.py", line 230, in get
net = instantiate_model(model_name, arch_params, checkpoint_num_classes, pretrained_weights, download_required_code)
File "super-gradients/src/super_gradients/training/models/model_factory.py", line 155, in instantiate_model
net = architecture_cls(arch_params=arch_params)
File "super-gradients/src/super_gradients/training/models/detection_models/pp_yolo_e/pp_yolo_e.py", line 439, in __init__
super().__init__(arch_params)
File "super-gradients/src/super_gradients/training/models/detection_models/pp_yolo_e/pp_yolo_e.py", line 108, in __init__
self.backbone = CSPResNetBackbone(**arch_params["backbone"], depth_mult=arch_params["depth_mult"], width_mult=arch_params["width_mult"])
File "super-gradients/src/super_gradients/common/decorators/factory_decorator.py", line 36, in wrapper
return func(*args, **kwargs)
File "super-gradients/src/super_gradients/training/models/detection_models/csp_resnet.py", line 221, in __init__
self.load_state_dict(state_dict)
File "anaconda3/envs/super-gradients/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2593, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for CSPResNetBackbone:
Missing key(s) in state_dict: "stem.conv1.seq.conv.weight", "stem.conv1.seq.bn.weight", "stem.conv1.seq.bn.bias", "stem.conv1.seq.bn.running_mean", "stem.conv1.seq.bn.running_var", "stem.conv2.seq.conv.weight", "stem.conv2.seq.bn.weight", "stem.conv2.seq.bn.bias", "stem.conv2.seq.bn.running_mean", "stem.conv2.seq.bn.running_var", "stem.conv3.seq.conv.weight", "stem.conv3.seq.bn.weight", "stem.conv3.seq.bn.bias", "stem.conv3.seq.bn.running_mean", "stem.conv3.seq.bn.running_var", "stages.0.conv_down.seq.conv.weight", "stages.0.conv_down.seq.bn.weight", "stages.0.conv_down.seq.bn.bias", "stages.0.conv_down.seq.bn.running_mean", "stages.0.conv_down.seq.bn.running_var", "stages.0.conv1.seq.conv.weight", "stages.0.conv1.seq.bn.weight", "stages.0.conv1.seq.bn.bias", "stages.0.conv1.seq.bn.running_mean", "stages.0.conv1.seq.bn.running_var", "stages.0.conv2.seq.conv.weight", "stages.0.conv2.seq.bn.weight", "stages.0.conv2.seq.bn.bias", "stages.0.conv2.seq.bn.running_mean", "stages.0.conv2.seq.bn.running_var", "stages.0.blocks.0.conv1.seq.conv.weight", "stages.0.blocks.0.conv1.seq.bn.weight", "stages.0.blocks.0.conv1.seq.bn.bias", "stages.0.blocks.0.conv1.seq.bn.running_mean", "stages.0.blocks.0.conv1.seq.bn.running_var", "stages.0.blocks.0.conv2.branch_3x3.conv.weight", "stages.0.blocks.0.conv2.branch_3x3.bn.weight", "stages.0.blocks.0.conv2.branch_3x3.bn.bias", "stages.0.blocks.0.conv2.branch_3x3.bn.running_mean", "stages.0.blocks.0.conv2.branch_3x3.bn.running_var", "stages.0.blocks.0.conv2.branch_1x1.conv.weight", "stages.0.blocks.0.conv2.branch_1x1.bn.weight", "stages.0.blocks.0.conv2.branch_1x1.bn.bias", "stages.0.blocks.0.conv2.branch_1x1.bn.running_mean", "stages.0.blocks.0.conv2.branch_1x1.bn.running_var", "stages.0.blocks.1.conv1.seq.conv.weight", "stages.0.blocks.1.conv1.seq.bn.weight", "stages.0.blocks.1.conv1.seq.bn.bias", "stages.0.blocks.1.conv1.seq.bn.running_mean", "stages.0.blocks.1.conv1.seq.bn.running_var", "stages.0.blocks.1.conv2.branch_3x3.conv.weight", "stages.0.blocks.1.conv2.branch_3x3.bn.weight", "stages.0.blocks.1.conv2.branch_3x3.bn.bias", "stages.0.blocks.1.conv2.branch_3x3.bn.running_mean", "stages.0.blocks.1.conv2.branch_3x3.bn.running_var", "stages.0.blocks.1.conv2.branch_1x1.conv.weight", "stages.0.blocks.1.conv2.branch_1x1.bn.weight", "stages.0.blocks.1.conv2.branch_1x1.bn.bias", "stages.0.blocks.1.conv2.branch_1x1.bn.running_mean", "stages.0.blocks.1.conv2.branch_1x1.bn.running_var", "stages.0.blocks.2.conv1.seq.conv.weight", "stages.0.blocks.2.conv1.seq.bn.weight", "stages.0.blocks.2.conv1.seq.bn.bias", "stages.0.blocks.2.conv1.seq.bn.running_mean", "stages.0.blocks.2.conv1.seq.bn.running_var", "stages.0.blocks.2.conv2.branch_3x3.conv.weight", "stages.0.blocks.2.conv2.branch_3x3.bn.weight", "stages.0.blocks.2.conv2.branch_3x3.bn.bias", "stages.0.blocks.2.conv2.branch_3x3.bn.running_mean", "stages.0.blocks.2.conv2.branch_3x3.bn.running_var", "stages.0.blocks.2.conv2.branch_1x1.conv.weight", "stages.0.blocks.2.conv2.branch_1x1.bn.weight", "stages.0.blocks.2.conv2.branch_1x1.bn.bias", "stages.0.blocks.2.conv2.branch_1x1.bn.running_mean", "stages.0.blocks.2.conv2.branch_1x1.bn.running_var", "stages.0.blocks.3.conv1.seq.conv.weight", "stages.0.blocks.3.conv1.seq.bn.weight", "stages.0.blocks.3.conv1.seq.bn.bias", "stages.0.blocks.3.conv1.seq.bn.running_mean", "stages.0.blocks.3.conv1.seq.bn.running_var", "stages.0.blocks.3.conv2.branch_3x3.conv.weight", "stages.0.blocks.3.conv2.branch_3x3.bn.weight", "stages.0.blocks.3.conv2.branch_3x3.bn.bias", "stages.0.blocks.3.conv2.branch_3x3.bn.running_mean", "stages.0.blocks.3.conv2.branch_3x3.bn.running_var", "stages.0.blocks.3.conv2.branch_1x1.conv.weight", "stages.0.blocks.3.conv2.branch_1x1.bn.weight", "stages.0.blocks.3.conv2.branch_1x1.bn.bias", "stages.0.blocks.3.conv2.branch_1x1.bn.running_mean", "stages.0.blocks.3.conv2.branch_1x1.bn.running_var", "stages.0.attn.project.weight", "stages.0.attn.project.bias", "stages.0.conv3.seq.conv.weight", "stages.0.conv3.seq.bn.weight", "stages.0.conv3.seq.bn.bias", "stages.0.conv3.seq.bn.running_mean", "stages.0.conv3.seq.bn.running_var", "stages.1.conv_down.seq.conv.weight", "stages.1.conv_down.seq.bn.weight", "stages.1.conv_down.seq.bn.bias", "stages.1.conv_down.seq.bn.running_mean", "stages.1.conv_down.seq.bn.running_var", "stages.1.conv1.seq.conv.weight", "stages.1.conv1.seq.bn.weight", "stages.1.conv1.seq.bn.bias", "stages.1.conv1.seq.bn.running_mean", "stages.1.conv1.seq.bn.running_var", "stages.1.conv2.seq.conv.weight", "stages.1.conv2.seq.bn.weight", "stages.1.conv2.seq.bn.bias", "stages.1.conv2.seq.bn.running_mean", "stages.1.conv2.seq.bn.running_var", "stages.1.blocks.0.conv1.seq.conv.weight", "stages.1.blocks.0.conv1.seq.bn.weight", "stages.1.blocks.0.conv1.seq.bn.bias", "stages.1.blocks.0.conv1.seq.bn.running_mean", "stages.1.blocks.0.conv1.seq.bn.running_var", "stages.1.blocks.0.conv2.branch_3x3.conv.weight", "stages.1.blocks.0.conv2.branch_3x3.bn.weight", "stages.1.blocks.0.conv2.branch_3x3.bn.bias", "stages.1.blocks.0.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.0.conv2.branch_3x3.bn.running_var", "stages.1.blocks.0.conv2.branch_1x1.conv.weight", "stages.1.blocks.0.conv2.branch_1x1.bn.weight", "stages.1.blocks.0.conv2.branch_1x1.bn.bias", "stages.1.blocks.0.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.0.conv2.branch_1x1.bn.running_var", "stages.1.blocks.1.conv1.seq.conv.weight", "stages.1.blocks.1.conv1.seq.bn.weight", "stages.1.blocks.1.conv1.seq.bn.bias", "stages.1.blocks.1.conv1.seq.bn.running_mean", "stages.1.blocks.1.conv1.seq.bn.running_var", "stages.1.blocks.1.conv2.branch_3x3.conv.weight", "stages.1.blocks.1.conv2.branch_3x3.bn.weight", "stages.1.blocks.1.conv2.branch_3x3.bn.bias", "stages.1.blocks.1.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.1.conv2.branch_3x3.bn.running_var", "stages.1.blocks.1.conv2.branch_1x1.conv.weight", "stages.1.blocks.1.conv2.branch_1x1.bn.weight", "stages.1.blocks.1.conv2.branch_1x1.bn.bias", "stages.1.blocks.1.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.1.conv2.branch_1x1.bn.running_var", "stages.1.blocks.2.conv1.seq.conv.weight", "stages.1.blocks.2.conv1.seq.bn.weight", "stages.1.blocks.2.conv1.seq.bn.bias", "stages.1.blocks.2.conv1.seq.bn.running_mean", "stages.1.blocks.2.conv1.seq.bn.running_var", "stages.1.blocks.2.conv2.branch_3x3.conv.weight", "stages.1.blocks.2.conv2.branch_3x3.bn.weight", "stages.1.blocks.2.conv2.branch_3x3.bn.bias", "stages.1.blocks.2.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.2.conv2.branch_3x3.bn.running_var", "stages.1.blocks.2.conv2.branch_1x1.conv.weight", "stages.1.blocks.2.conv2.branch_1x1.bn.weight", "stages.1.blocks.2.conv2.branch_1x1.bn.bias", "stages.1.blocks.2.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.2.conv2.branch_1x1.bn.running_var", "stages.1.blocks.3.conv1.seq.conv.weight", "stages.1.blocks.3.conv1.seq.bn.weight", "stages.1.blocks.3.conv1.seq.bn.bias", "stages.1.blocks.3.conv1.seq.bn.running_mean", "stages.1.blocks.3.conv1.seq.bn.running_var", "stages.1.blocks.3.conv2.branch_3x3.conv.weight", "stages.1.blocks.3.conv2.branch_3x3.bn.weight", "stages.1.blocks.3.conv2.branch_3x3.bn.bias", "stages.1.blocks.3.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.3.conv2.branch_3x3.bn.running_var", "stages.1.blocks.3.conv2.branch_1x1.conv.weight", "stages.1.blocks.3.conv2.branch_1x1.bn.weight", "stages.1.blocks.3.conv2.branch_1x1.bn.bias", "stages.1.blocks.3.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.3.conv2.branch_1x1.bn.running_var", "stages.1.blocks.4.conv1.seq.conv.weight", "stages.1.blocks.4.conv1.seq.bn.weight", "stages.1.blocks.4.conv1.seq.bn.bias", "stages.1.blocks.4.conv1.seq.bn.running_mean", "stages.1.blocks.4.conv1.seq.bn.running_var", "stages.1.blocks.4.conv2.branch_3x3.conv.weight", "stages.1.blocks.4.conv2.branch_3x3.bn.weight", "stages.1.blocks.4.conv2.branch_3x3.bn.bias", "stages.1.blocks.4.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.4.conv2.branch_3x3.bn.running_var", "stages.1.blocks.4.conv2.branch_1x1.conv.weight", "stages.1.blocks.4.conv2.branch_1x1.bn.weight", "stages.1.blocks.4.conv2.branch_1x1.bn.bias", "stages.1.blocks.4.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.4.conv2.branch_1x1.bn.running_var", "stages.1.blocks.5.conv1.seq.conv.weight", "stages.1.blocks.5.conv1.seq.bn.weight", "stages.1.blocks.5.conv1.seq.bn.bias", "stages.1.blocks.5.conv1.seq.bn.running_mean", "stages.1.blocks.5.conv1.seq.bn.running_var", "stages.1.blocks.5.conv2.branch_3x3.conv.weight", "stages.1.blocks.5.conv2.branch_3x3.bn.weight", "stages.1.blocks.5.conv2.branch_3x3.bn.bias", "stages.1.blocks.5.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.5.conv2.branch_3x3.bn.running_var", "stages.1.blocks.5.conv2.branch_1x1.conv.weight", "stages.1.blocks.5.conv2.branch_1x1.bn.weight", "stages.1.blocks.5.conv2.branch_1x1.bn.bias", "stages.1.blocks.5.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.5.conv2.branch_1x1.bn.running_var", "stages.1.blocks.6.conv1.seq.conv.weight", "stages.1.blocks.6.conv1.seq.bn.weight", "stages.1.blocks.6.conv1.seq.bn.bias", "stages.1.blocks.6.conv1.seq.bn.running_mean", "stages.1.blocks.6.conv1.seq.bn.running_var", "stages.1.blocks.6.conv2.branch_3x3.conv.weight", "stages.1.blocks.6.conv2.branch_3x3.bn.weight", "stages.1.blocks.6.conv2.branch_3x3.bn.bias", "stages.1.blocks.6.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.6.conv2.branch_3x3.bn.running_var", "stages.1.blocks.6.conv2.branch_1x1.conv.weight", "stages.1.blocks.6.conv2.branch_1x1.bn.weight", "stages.1.blocks.6.conv2.branch_1x1.bn.bias", "stages.1.blocks.6.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.6.conv2.branch_1x1.bn.running_var", "stages.1.blocks.7.conv1.seq.conv.weight", "stages.1.blocks.7.conv1.seq.bn.weight", "stages.1.blocks.7.conv1.seq.bn.bias", "stages.1.blocks.7.conv1.seq.bn.running_mean", "stages.1.blocks.7.conv1.seq.bn.running_var", "stages.1.blocks.7.conv2.branch_3x3.conv.weight", "stages.1.blocks.7.conv2.branch_3x3.bn.weight", "stages.1.blocks.7.conv2.branch_3x3.bn.bias", "stages.1.blocks.7.conv2.branch_3x3.bn.running_mean", "stages.1.blocks.7.conv2.branch_3x3.bn.running_var", "stages.1.blocks.7.conv2.branch_1x1.conv.weight", "stages.1.blocks.7.conv2.branch_1x1.bn.weight", "stages.1.blocks.7.conv2.branch_1x1.bn.bias", "stages.1.blocks.7.conv2.branch_1x1.bn.running_mean", "stages.1.blocks.7.conv2.branch_1x1.bn.running_var", "stages.1.attn.project.weight", "stages.1.attn.project.bias", "stages.1.conv3.seq.conv.weight", "stages.1.conv3.seq.bn.weight", "stages.1.conv3.seq.bn.bias", "stages.1.conv3.seq.bn.running_mean", "stages.1.conv3.seq.bn.running_var", "stages.2.conv_down.seq.conv.weight", "stages.2.conv_down.seq.bn.weight", "stages.2.conv_down.seq.bn.bias", "stages.2.conv_down.seq.bn.running_mean", "stages.2.conv_down.seq.bn.running_var", "stages.2.conv1.seq.conv.weight", "stages.2.conv1.seq.bn.weight", "stages.2.conv1.seq.bn.bias", "stages.2.conv1.seq.bn.running_mean", "stages.2.conv1.seq.bn.running_var", "stages.2.conv2.seq.conv.weight", "stages.2.conv2.seq.bn.weight", "stages.2.conv2.seq.bn.bias", "stages.2.conv2.seq.bn.running_mean", "stages.2.conv2.seq.bn.running_var", "stages.2.blocks.0.conv1.seq.conv.weight", "stages.2.blocks.0.conv1.seq.bn.weight", "stages.2.blocks.0.conv1.seq.bn.bias", "stages.2.blocks.0.conv1.seq.bn.running_mean", "stages.2.blocks.0.conv1.seq.bn.running_var", "stages.2.blocks.0.conv2.branch_3x3.conv.weight", "stages.2.blocks.0.conv2.branch_3x3.bn.weight", "stages.2.blocks.0.conv2.branch_3x3.bn.bias", "stages.2.blocks.0.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.0.conv2.branch_3x3.bn.running_var", "stages.2.blocks.0.conv2.branch_1x1.conv.weight", "stages.2.blocks.0.conv2.branch_1x1.bn.weight", "stages.2.blocks.0.conv2.branch_1x1.bn.bias", "stages.2.blocks.0.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.0.conv2.branch_1x1.bn.running_var", "stages.2.blocks.1.conv1.seq.conv.weight", "stages.2.blocks.1.conv1.seq.bn.weight", "stages.2.blocks.1.conv1.seq.bn.bias", "stages.2.blocks.1.conv1.seq.bn.running_mean", "stages.2.blocks.1.conv1.seq.bn.running_var", "stages.2.blocks.1.conv2.branch_3x3.conv.weight", "stages.2.blocks.1.conv2.branch_3x3.bn.weight", "stages.2.blocks.1.conv2.branch_3x3.bn.bias", "stages.2.blocks.1.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.1.conv2.branch_3x3.bn.running_var", "stages.2.blocks.1.conv2.branch_1x1.conv.weight", "stages.2.blocks.1.conv2.branch_1x1.bn.weight", "stages.2.blocks.1.conv2.branch_1x1.bn.bias", "stages.2.blocks.1.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.1.conv2.branch_1x1.bn.running_var", "stages.2.blocks.2.conv1.seq.conv.weight", "stages.2.blocks.2.conv1.seq.bn.weight", "stages.2.blocks.2.conv1.seq.bn.bias", "stages.2.blocks.2.conv1.seq.bn.running_mean", "stages.2.blocks.2.conv1.seq.bn.running_var", "stages.2.blocks.2.conv2.branch_3x3.conv.weight", "stages.2.blocks.2.conv2.branch_3x3.bn.weight", "stages.2.blocks.2.conv2.branch_3x3.bn.bias", "stages.2.blocks.2.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.2.conv2.branch_3x3.bn.running_var", "stages.2.blocks.2.conv2.branch_1x1.conv.weight", "stages.2.blocks.2.conv2.branch_1x1.bn.weight", "stages.2.blocks.2.conv2.branch_1x1.bn.bias", "stages.2.blocks.2.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.2.conv2.branch_1x1.bn.running_var", "stages.2.blocks.3.conv1.seq.conv.weight", "stages.2.blocks.3.conv1.seq.bn.weight", "stages.2.blocks.3.conv1.seq.bn.bias", "stages.2.blocks.3.conv1.seq.bn.running_mean", "stages.2.blocks.3.conv1.seq.bn.running_var", "stages.2.blocks.3.conv2.branch_3x3.conv.weight", "stages.2.blocks.3.conv2.branch_3x3.bn.weight", "stages.2.blocks.3.conv2.branch_3x3.bn.bias", "stages.2.blocks.3.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.3.conv2.branch_3x3.bn.running_var", "stages.2.blocks.3.conv2.branch_1x1.conv.weight", "stages.2.blocks.3.conv2.branch_1x1.bn.weight", "stages.2.blocks.3.conv2.branch_1x1.bn.bias", "stages.2.blocks.3.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.3.conv2.branch_1x1.bn.running_var", "stages.2.blocks.4.conv1.seq.conv.weight", "stages.2.blocks.4.conv1.seq.bn.weight", "stages.2.blocks.4.conv1.seq.bn.bias", "stages.2.blocks.4.conv1.seq.bn.running_mean", "stages.2.blocks.4.conv1.seq.bn.running_var", "stages.2.blocks.4.conv2.branch_3x3.conv.weight", "stages.2.blocks.4.conv2.branch_3x3.bn.weight", "stages.2.blocks.4.conv2.branch_3x3.bn.bias", "stages.2.blocks.4.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.4.conv2.branch_3x3.bn.running_var", "stages.2.blocks.4.conv2.branch_1x1.conv.weight", "stages.2.blocks.4.conv2.branch_1x1.bn.weight", "stages.2.blocks.4.conv2.branch_1x1.bn.bias", "stages.2.blocks.4.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.4.conv2.branch_1x1.bn.running_var", "stages.2.blocks.5.conv1.seq.conv.weight", "stages.2.blocks.5.conv1.seq.bn.weight", "stages.2.blocks.5.conv1.seq.bn.bias", "stages.2.blocks.5.conv1.seq.bn.running_mean", "stages.2.blocks.5.conv1.seq.bn.running_var", "stages.2.blocks.5.conv2.branch_3x3.conv.weight", "stages.2.blocks.5.conv2.branch_3x3.bn.weight", "stages.2.blocks.5.conv2.branch_3x3.bn.bias", "stages.2.blocks.5.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.5.conv2.branch_3x3.bn.running_var", "stages.2.blocks.5.conv2.branch_1x1.conv.weight", "stages.2.blocks.5.conv2.branch_1x1.bn.weight", "stages.2.blocks.5.conv2.branch_1x1.bn.bias", "stages.2.blocks.5.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.5.conv2.branch_1x1.bn.running_var", "stages.2.blocks.6.conv1.seq.conv.weight", "stages.2.blocks.6.conv1.seq.bn.weight", "stages.2.blocks.6.conv1.seq.bn.bias", "stages.2.blocks.6.conv1.seq.bn.running_mean", "stages.2.blocks.6.conv1.seq.bn.running_var", "stages.2.blocks.6.conv2.branch_3x3.conv.weight", "stages.2.blocks.6.conv2.branch_3x3.bn.weight", "stages.2.blocks.6.conv2.branch_3x3.bn.bias", "stages.2.blocks.6.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.6.conv2.branch_3x3.bn.running_var", "stages.2.blocks.6.conv2.branch_1x1.conv.weight", "stages.2.blocks.6.conv2.branch_1x1.bn.weight", "stages.2.blocks.6.conv2.branch_1x1.bn.bias", "stages.2.blocks.6.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.6.conv2.branch_1x1.bn.running_var", "stages.2.blocks.7.conv1.seq.conv.weight", "stages.2.blocks.7.conv1.seq.bn.weight", "stages.2.blocks.7.conv1.seq.bn.bias", "stages.2.blocks.7.conv1.seq.bn.running_mean", "stages.2.blocks.7.conv1.seq.bn.running_var", "stages.2.blocks.7.conv2.branch_3x3.conv.weight", "stages.2.blocks.7.conv2.branch_3x3.bn.weight", "stages.2.blocks.7.conv2.branch_3x3.bn.bias", "stages.2.blocks.7.conv2.branch_3x3.bn.running_mean", "stages.2.blocks.7.conv2.branch_3x3.bn.running_var", "stages.2.blocks.7.conv2.branch_1x1.conv.weight", "stages.2.blocks.7.conv2.branch_1x1.bn.weight", "stages.2.blocks.7.conv2.branch_1x1.bn.bias", "stages.2.blocks.7.conv2.branch_1x1.bn.running_mean", "stages.2.blocks.7.conv2.branch_1x1.bn.running_var", "stages.2.attn.project.weight", "stages.2.attn.project.bias", "stages.2.conv3.seq.conv.weight", "stages.2.conv3.seq.bn.weight", "stages.2.conv3.seq.bn.bias", "stages.2.conv3.seq.bn.running_mean", "stages.2.conv3.seq.bn.running_var", "stages.3.conv_down.seq.conv.weight", "stages.3.conv_down.seq.bn.weight", "stages.3.conv_down.seq.bn.bias", "stages.3.conv_down.seq.bn.running_mean", "stages.3.conv_down.seq.bn.running_var", "stages.3.conv1.seq.conv.weight", "stages.3.conv1.seq.bn.weight", "stages.3.conv1.seq.bn.bias", "stages.3.conv1.seq.bn.running_mean", "stages.3.conv1.seq.bn.running_var", "stages.3.conv2.seq.conv.weight", "stages.3.conv2.seq.bn.weight", "stages.3.conv2.seq.bn.bias", "stages.3.conv2.seq.bn.running_mean", "stages.3.conv2.seq.bn.running_var", "stages.3.blocks.0.conv1.seq.conv.weight", "stages.3.blocks.0.conv1.seq.bn.weight", "stages.3.blocks.0.conv1.seq.bn.bias", "stages.3.blocks.0.conv1.seq.bn.running_mean", "stages.3.blocks.0.conv1.seq.bn.running_var", "stages.3.blocks.0.conv2.branch_3x3.conv.weight", "stages.3.blocks.0.conv2.branch_3x3.bn.weight", "stages.3.blocks.0.conv2.branch_3x3.bn.bias", "stages.3.blocks.0.conv2.branch_3x3.bn.running_mean", "stages.3.blocks.0.conv2.branch_3x3.bn.running_var", "stages.3.blocks.0.conv2.branch_1x1.conv.weight", "stages.3.blocks.0.conv2.branch_1x1.bn.weight", "stages.3.blocks.0.conv2.branch_1x1.bn.bias", "stages.3.blocks.0.conv2.branch_1x1.bn.running_mean", "stages.3.blocks.0.conv2.branch_1x1.bn.running_var", "stages.3.blocks.1.conv1.seq.conv.weight", "stages.3.blocks.1.conv1.seq.bn.weight", "stages.3.blocks.1.conv1.seq.bn.bias", "stages.3.blocks.1.conv1.seq.bn.running_mean", "stages.3.blocks.1.conv1.seq.bn.running_var", "stages.3.blocks.1.conv2.branch_3x3.conv.weight", "stages.3.blocks.1.conv2.branch_3x3.bn.weight", "stages.3.blocks.1.conv2.branch_3x3.bn.bias", "stages.3.blocks.1.conv2.branch_3x3.bn.running_mean", "stages.3.blocks.1.conv2.branch_3x3.bn.running_var", "stages.3.blocks.1.conv2.branch_1x1.conv.weight", "stages.3.blocks.1.conv2.branch_1x1.bn.weight", "stages.3.blocks.1.conv2.branch_1x1.bn.bias", "stages.3.blocks.1.conv2.branch_1x1.bn.running_mean", "stages.3.blocks.1.conv2.branch_1x1.bn.running_var", "stages.3.blocks.2.conv1.seq.conv.weight", "stages.3.blocks.2.conv1.seq.bn.weight", "stages.3.blocks.2.conv1.seq.bn.bias", "stages.3.blocks.2.conv1.seq.bn.running_mean", "stages.3.blocks.2.conv1.seq.bn.running_var", "stages.3.blocks.2.conv2.branch_3x3.conv.weight", "stages.3.blocks.2.conv2.branch_3x3.bn.weight", "stages.3.blocks.2.conv2.branch_3x3.bn.bias", "stages.3.blocks.2.conv2.branch_3x3.bn.running_mean", "stages.3.blocks.2.conv2.branch_3x3.bn.running_var", "stages.3.blocks.2.conv2.branch_1x1.conv.weight", "stages.3.blocks.2.conv2.branch_1x1.bn.weight", "stages.3.blocks.2.conv2.branch_1x1.bn.bias", "stages.3.blocks.2.conv2.branch_1x1.bn.running_mean", "stages.3.blocks.2.conv2.branch_1x1.bn.running_var", "stages.3.blocks.3.conv1.seq.conv.weight", "stages.3.blocks.3.conv1.seq.bn.weight", "stages.3.blocks.3.conv1.seq.bn.bias", "stages.3.blocks.3.conv1.seq.bn.running_mean", "stages.3.blocks.3.conv1.seq.bn.running_var", "stages.3.blocks.3.conv2.branch_3x3.conv.weight", "stages.3.blocks.3.conv2.branch_3x3.bn.weight", "stages.3.blocks.3.conv2.branch_3x3.bn.bias", "stages.3.blocks.3.conv2.branch_3x3.bn.running_mean", "stages.3.blocks.3.conv2.branch_3x3.bn.running_var", "stages.3.blocks.3.conv2.branch_1x1.conv.weight", "stages.3.blocks.3.conv2.branch_1x1.bn.weight", "stages.3.blocks.3.conv2.branch_1x1.bn.bias", "stages.3.blocks.3.conv2.branch_1x1.bn.running_mean", "stages.3.blocks.3.conv2.branch_1x1.bn.running_var", "stages.3.attn.project.weight", "stages.3.attn.project.bias", "stages.3.conv3.seq.conv.weight", "stages.3.conv3.seq.bn.weight", "stages.3.conv3.seq.bn.bias", "stages.3.conv3.seq.bn.running_mean", "stages.3.conv3.seq.bn.running_var".
Unexpected key(s) in state_dict: "net", "acc", "epoch", "optimizer_state_dict", "scaler_state_dict", "ema_net".
Versions
No response