Skip to content

error while running of pre trained model #12

@Kafka2122

Description

@Kafka2122

Hello, I am trying to use the pretrained models given in the readme. I am using the nuscene(v.1-0mini) dataset and the MinkUNet34 pretrained model. I first run the command

python convert_models.py —downstream —ckpt path_to_downstream_checkpoint

and then the command

python eval.py —split val —config path_to_downstream_model/config.yaml —ckpt path_to_downstream_checkpoint/trained_model_XXX.ckpt

The first command works fine but the second command gives an error :

Traceback (most recent call last):
File "/root/docker_data/ALSO2/ALSO/also_selfsup/eval.py", line 78, in
net.load_state_dict(torch.load(opts.ckpt), strict=True)
File "/root/miniconda3/envs/env2/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1671, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for MinkUNet34:
Missing key(s) in state_dict: "stem.0.kernel", "stem.1.weight", "stem.1.bias", "stem.1.running_mean", "stem.1.running_var", "stage1.0.net.0.kernel", "stage1.0.net.1.weight", "stage1.0.net.1.bias", "stage1.0.net.1.running_mean", "stage1.0.net.1.running_var", "stage1.1.net.0.kernel", "stage1.1.net.1.weight", "stage1.1.net.1.bias", "stage1.1.net.1.running_mean", "stage1.1.net.1.running_var", "stage1.1.net.3.kernel", "stage1.1.net.4.weight", "stage1.1.net.4.bias", "stage1.1.net.4.running_mean", "stage1.1.net.4.running_var", "stage1.2.net.0.kernel", "stage1.2.net.1.weight", "stage1.2.net.1.bias", "stage1.2.net.1.running_mean", "stage1.2.net.1.running_var", "stage1.2.net.3.kernel", "stage1.2.net.4.weight", "stage1.2.net.4.bias", "stage1.2.net.4.running_mean", "stage1.2.net.4.running_var", "stage2.0.net.0.kernel", "stage2.0.net.1.weight", "stage2.0.net.1.bias", "stage2.0.net.1.running_mean", "stage2.0.net.1.running_var", "stage2.1.net.0.kernel", "stage2.1.net.1.weight", "stage2.1.net.1.bias", "stage2.1.net.1.running_mean", "stage2.1.net.1.running_var", "stage2.1.net.3.kernel", "stage2.1.net.4.weight", "stage2.1.net.4.bias", "stage2.1.net.4.running_mean", "stage2.1.net.4.running_var", "stage2.1.downsample.0.kernel", "stage2.1.downsample.1.weight", "stage2.1.downsample.1.bias", "stage2.1.downsample.1.running_mean", "stage2.1.downsample.1.running_var", "stage2.2.net.0.kernel", "stage2.2.net.1.weight", "stage2.2.net.1.bias", "stage2.2.net.1.running_mean", "stage2.2.net.1.running_var", "stage2.2.net.3.kernel", "stage2.2.net.4.weight", "stage2.2.net.4.bias", "stage2.2.net.4.running_mean", "stage2.2.net.4.running_var", "stage2.3.net.0.kernel", "stage2.3.net.1.weight", "stage2.3.net.1.bias", "stage2.3.net.1.running_mean", "stage2.3.net.1.running_var", "stage2.3.net.3.kernel", "stage2.3.net.4.weight", "stage2.3.net.4.bias", "stage2.3.net.4.running_mean", "stage2.3.net.4.running_var", "stage3.0.net.0.kernel", "stage3.0.net.1.weight", "stage3.0.net.1.bias", "stage3.0.net.1.running_mean", "stage3.0.net.1.running_var", "stage3.1.net.0.kernel", "stage3.1.net.1.weight", "stage3.1.net.1.bias", "stage3.1.net.1.running_mean", "stage3.1.net.1.running_var", "stage3.1.net.3.kernel", "stage3.1.net.4.weight", "stage3.1.net.4.bias", "stage3.1.net.4.running_mean", "stage3.1.net.4.running_var", "stage3.1.downsample.0.kernel", "stage3.1.downsample.1.weight", "stage3.1.downsample.1.bias", "stage3.1.downsample.1.running_mean", "stage3.1.downsample.1.running_var", "stage3.2.net.0.kernel", "stage3.2.net.1.weight", "stage3.2.net.1.bias", "stage3.2.net.1.running_mean", "stage3.2.net.1.running_var", "stage3.2.net.3.kernel", "stage3.2.net.4.weight", "stage3.2.net.4.bias", "stage3.2.net.4.running_mean", "stage3.2.net.4.running_var", "stage3.3.net.0.kernel", "stage3.3.net.1.weight", "stage3.3.net.1.bias", "stage3.3.net.1.running_mean", "stage3.3.net.1.running_var", "stage3.3.net.3.kernel", "stage3.3.net.4.weight", "stage3.3.net.4.bias", "stage3.3.net.4.running_mean", "stage3.3.net.4.running_var", "stage3.4.net.0.kernel", "stage3.4.net.1.weight", "stage3.4.net.1.bias", "stage3.4.net.1.running_mean", "stage3.4.net.1.running_var", "stage3.4.net.3.kernel", "stage3.4.net.4.weight", "stage3.4.net.4.bias", "stage3.4.net.4.running_mean", "stage3.4.net.4.running_var", "stage4.0.net.0.kernel", "stage4.0.net.1.weight", "stage4.0.net.1.bias", "stage4.0.net.1.running_mean", "stage4.0.net.1.running_var", "stage4.1.net.0.kernel", "stage4.1.net.1.weight", "stage4.1.net.1.bias", "stage4.1.net.1.running_mean", "stage4.1.net.1.running_var", "stage4.1.net.3.kernel", "stage4.1.net.4.weight", "stage4.1.net.4.bias", "stage4.1.net.4.running_mean", "stage4.1.net.4.running_var", "stage4.1.downsample.0.kernel", "stage4.1.downsample.1.weight", "stage4.1.downsample.1.bias", "stage4.1.downsample.1.running_mean", "stage4.1.downsample.1.running_var", "stage4.2.net.0.kernel", "stage4.2.net.1.weight", "stage4.2.net.1.bias", "stage4.2.net.1.running_mean", "stage4.2.net.1.running_var", "stage4.2.net.3.kernel", "stage4.2.net.4.weight", "stage4.2.net.4.bias", "stage4.2.net.4.running_mean", "stage4.2.net.4.running_var", "stage4.3.net.0.kernel", "stage4.3.net.1.weight", "stage4.3.net.1.bias", "stage4.3.net.1.running_mean", "stage4.3.net.1.running_var", "stage4.3.net.3.kernel", "stage4.3.net.4.weight", "stage4.3.net.4.bias", "stage4.3.net.4.running_mean", "stage4.3.net.4.running_var", "stage4.4.net.0.kernel", "stage4.4.net.1.weight", "stage4.4.net.1.bias", "stage4.4.net.1.running_mean", "stage4.4.net.1.running_var", "stage4.4.net.3.kernel", "stage4.4.net.4.weight", "stage4.4.net.4.bias", "stage4.4.net.4.running_mean", "stage4.4.net.4.running_var", "stage4.5.net.0.kernel", "stage4.5.net.1.weight", "stage4.5.net.1.bias", "stage4.5.net.1.running_mean", "stage4.5.net.1.running_var", "stage4.5.net.3.kernel", "stage4.5.net.4.weight", "stage4.5.net.4.bias", "stage4.5.net.4.running_mean", "stage4.5.net.4.running_var", "stage4.6.net.0.kernel", "stage4.6.net.1.weight", "stage4.6.net.1.bias", "stage4.6.net.1.running_mean", "stage4.6.net.1.running_var", "stage4.6.net.3.kernel", "stage4.6.net.4.weight", "stage4.6.net.4.bias", "stage4.6.net.4.running_mean", "stage4.6.net.4.running_var", "up1.0.net.0.kernel", "up1.0.net.1.weight", "up1.0.net.1.bias", "up1.0.net.1.running_mean", "up1.0.net.1.running_var", "up1.1.0.net.0.kernel", "up1.1.0.net.1.weight", "up1.1.0.net.1.bias", "up1.1.0.net.1.running_mean", "up1.1.0.net.1.running_var", "up1.1.0.net.3.kernel", "up1.1.0.net.4.weight", "up1.1.0.net.4.bias", "up1.1.0.net.4.running_mean", "up1.1.0.net.4.running_var", "up1.1.0.downsample.0.kernel", "up1.1.0.downsample.1.weight", "up1.1.0.downsample.1.bias", "up1.1.0.downsample.1.running_mean", "up1.1.0.downsample.1.running_var", "up1.1.1.net.0.kernel", "up1.1.1.net.1.weight", "up1.1.1.net.1.bias", "up1.1.1.net.1.running_mean", "up1.1.1.net.1.running_var", "up1.1.1.net.3.kernel", "up1.1.1.net.4.weight", "up1.1.1.net.4.bias", "up1.1.1.net.4.running_mean", "up1.1.1.net.4.running_var", "up2.0.net.0.kernel", "up2.0.net.1.weight", "up2.0.net.1.bias", "up2.0.net.1.running_mean", "up2.0.net.1.running_var", "up2.1.0.net.0.kernel", "up2.1.0.net.1.weight", "up2.1.0.net.1.bias", "up2.1.0.net.1.running_mean", "up2.1.0.net.1.running_var", "up2.1.0.net.3.kernel", "up2.1.0.net.4.weight", "up2.1.0.net.4.bias", "up2.1.0.net.4.running_mean", "up2.1.0.net.4.running_var", "up2.1.0.downsample.0.kernel", "up2.1.0.downsample.1.weight", "up2.1.0.downsample.1.bias", "up2.1.0.downsample.1.running_mean", "up2.1.0.downsample.1.running_var", "up2.1.1.net.0.kernel", "up2.1.1.net.1.weight", "up2.1.1.net.1.bias", "up2.1.1.net.1.running_mean", "up2.1.1.net.1.running_var", "up2.1.1.net.3.kernel", "up2.1.1.net.4.weight", "up2.1.1.net.4.bias", "up2.1.1.net.4.running_mean", "up2.1.1.net.4.running_var", "up3.0.net.0.kernel", "up3.0.net.1.weight", "up3.0.net.1.bias", "up3.0.net.1.running_mean", "up3.0.net.1.running_var", "up3.1.0.net.0.kernel", "up3.1.0.net.1.weight", "up3.1.0.net.1.bias", "up3.1.0.net.1.running_mean", "up3.1.0.net.1.running_var", "up3.1.0.net.3.kernel", "up3.1.0.net.4.weight", "up3.1.0.net.4.bias", "up3.1.0.net.4.running_mean", "up3.1.0.net.4.running_var", "up3.1.0.downsample.0.kernel", "up3.1.0.downsample.1.weight", "up3.1.0.downsample.1.bias", "up3.1.0.downsample.1.running_mean", "up3.1.0.downsample.1.running_var", "up3.1.1.net.0.kernel", "up3.1.1.net.1.weight", "up3.1.1.net.1.bias", "up3.1.1.net.1.running_mean", "up3.1.1.net.1.running_var", "up3.1.1.net.3.kernel", "up3.1.1.net.4.weight", "up3.1.1.net.4.bias", "up3.1.1.net.4.running_mean", "up3.1.1.net.4.running_var", "up4.0.net.0.kernel", "up4.0.net.1.weight", "up4.0.net.1.bias", "up4.0.net.1.running_mean", "up4.0.net.1.running_var", "up4.1.0.net.0.kernel", "up4.1.0.net.1.weight", "up4.1.0.net.1.bias", "up4.1.0.net.1.running_mean", "up4.1.0.net.1.running_var", "up4.1.0.net.3.kernel", "up4.1.0.net.4.weight", "up4.1.0.net.4.bias", "up4.1.0.net.4.running_mean", "up4.1.0.net.4.running_var", "up4.1.0.downsample.0.kernel", "up4.1.0.downsample.1.weight", "up4.1.0.downsample.1.bias", "up4.1.0.downsample.1.running_mean", "up4.1.0.downsample.1.running_var", "up4.1.1.net.0.kernel", "up4.1.1.net.1.weight", "up4.1.1.net.1.bias", "up4.1.1.net.1.running_mean", "up4.1.1.net.1.running_var", "up4.1.1.net.3.kernel", "up4.1.1.net.4.weight", "up4.1.1.net.4.bias", "up4.1.1.net.4.running_mean", "up4.1.1.net.4.running_var", "classifier.0.weight", "classifier.0.bias".

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions