Description
Thanks for your error report and we appreciate it a lot.
Checklist
- There was a similar problem in unexpected key in source state_dict #1473, but the script was written only by timm and swin..
- The bug has not been fixed in the latest version.
Describe the bug
I'm trying to convert a model to ONNX format using tools/deploy.py, first checking the Mask2Former model initialization.
I extract the state_dict from my original checkpoint in .ckpt format and send it to the model.
However, I encounter the bug Unexpected key and missing keys. Also i met problem with "meta" key.
Reproduction
pl_module = Mask2Former.load_from_checkpoint('check.ckpt')
clean_state_dict = {k.replace("model.", ""): v for k, v in state_dict.items()}
torch.save(clean_state_dict, "check.pth")
model = init_model(config_path, "check.pth")
and got:
unexpected key in source state_dict: pixel_level_module.encoder.embeddings.patch_embeddings.projection.weight, pixel_level_module.encoder.embeddings.patch_embeddings.projection.bias, pixel_level_module.encoder.embeddings.norm.weight, pixel_level_module.encoder.embeddings.norm.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.0.blocks.0.output.dense.weight, pixel_level_module.encoder.encoder.layers.0.blocks.0.output.dense.bias, pixel_level_module.encoder.encoder.layers.0.blocks.1.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.0.blocks.1.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.0.blocks.1.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.0.blocks.1.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.0.blocks.1.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.0.blocks.1.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.0.blocks.1.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.7.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.7.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.7.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.7.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.7.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.7.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.7.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.7.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.7.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.7.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.7.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.8.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.8.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.9.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.9.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.10.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.10.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.11.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.11.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.12.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.12.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.13.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.13.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.14.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.14.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.15.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.15.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.16.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.16.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.2.blocks.17.output.dense.weight, pixel_level_module.encoder.encoder.layers.2.blocks.17.output.dense.bias, pixel_level_module.encoder.encoder.layers.2.downsample.reduction.weight, pixel_level_module.encoder.encoder.layers.2.downsample.norm.weight, pixel_level_module.encoder.encoder.layers.2.downsample.norm.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.3.blocks.0.output.dense.weight, pixel_level_module.encoder.encoder.layers.3.blocks.0.output.dense.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.layernorm_before.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.layernorm_before.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.relative_position_bias_table, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.relative_position_index, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.query.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.query.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.key.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.key.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.value.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.self.value.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.output.dense.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.attention.output.dense.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.layernorm_after.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.layernorm_after.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.intermediate.dense.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.intermediate.dense.bias, pixel_level_module.encoder.encoder.layers.3.blocks.1.output.dense.weight, pixel_level_module.encoder.encoder.layers.3.blocks.1.output.dense.bias, pixel_level_module.encoder.hidden_states_norms.stage1.weight, pixel_level_module.encoder.hidden_states_norms.stage1.bias, pixel_level_module.encoder.hidden_states_norms.stage2.weight, pixel_level_module.encoder.hidden_states_norms.stage2.bias, pixel_level_module.encoder.hidden_states_norms.stage3.weight, pixel_level_module.encoder.hidden_states_norms.stage3.bias, pixel_level_module.encoder.hidden_states_norms.stage4.weight, pixel_level_module.encoder.hidden_states_norms.stage4.bias, pixel_level_module.decoder.level_embed, pixel_level_module.decoder.input_projections.0.0.weight, pixel_level_module.decoder.input_projections.0.0.bias, pixel_level_module.decoder.input_projections.0.1.weight, pixel_level_module.decoder.input_projections.0.1.bias, pixel_level_module.decoder.input_projections.1.0.weight, pixel_level_module.decoder.input_projections.1.0.bias, pixel_level_module.decoder.input_projections.1.1.weight, pixel_level_module.decoder.input_projections.1.1.bias, pixel_level_module.decoder.input_projections.2.0.weight, pixel_level_module.decoder.input_projections.2.0.bias, pixel_level_module.decoder.input_projections.2.1.weight, pixel_level_module.decoder.input_projections.2.1.bias, pixel_level_module.decoder.encoder.layers.0.self_attn.sampling_offsets.weight, pixel_level_module.decoder.encoder.layers.0.self_attn.sampling_offsets.bias, pixel_level_module.decoder.encoder.layers.0.self_attn.attention_weights.weight, pixel_level_module.decoder.encoder.layers.0.self_attn.attention_weights.bias, pixel_level_module.decoder.encoder.layers.0.self_attn.value_proj.weight, pixel_level_module.decoder.encoder.layers.0.self_attn.value_proj.bias, pixel_level_module.decoder.encoder.layers.0.self_attn.output_proj.weight, pixel_level_module.decoder.encoder.layers.0.self_attn.output_proj.bias, pixel_level_module.decoder.encoder.layers.0.self_attn_layer_norm.weight, pixel_level_module.decoder.encoder.layers.0.self_attn_layer_norm.bias, pixel_level_module.decoder.encoder.layers.0.fc1.weight, pixel_level_module.decoder.encoder.layers.0.fc1.bias, pixel_level_module.decoder.encoder.layers.0.fc2.weight, pixel_level_module.decoder.encoder.layers.0.fc2.bias, pixel_level_module.decoder.encoder.layers.0.final_layer_norm.weight, pixel_level_module.decoder.encoder.layers.0.final_layer_norm.bias, pixel_level_module.decoder.encoder.layers.1.self_attn.sampling_offsets.weight, pixel_level_module.decoder.encoder.layers.1.self_attn.sampling_offsets.bias, pixel_level_module.decoder.encoder.layers.1.self_attn.attention_weights.weight, pixel_level_module.decoder.encoder.layers.1.self_attn.attention_weights.bias, pixel_level_module.decoder.encoder.layers.1.self_attn.value_proj.weight, pixel_level_module.decoder.encoder.layers.1.self_attn.value_proj.bias, pixel_level_module.decoder.encoder.layers.1.self_attn.output_proj.weight, pixel_level_module.decoder.encoder.layers.1.self_attn.output_proj.bias, pixel_level_module.decoder.encoder.layers.1.self_attn_layer_norm.weight, pixel_level_module.decoder.encoder.layers.1.self_attn_layer_norm.bias, pixel_level_module.decoder.encoder.layers.1.fc1.weight, pixel_level_module.decoder.encoder.layers.1.fc1.bias, pixel_level_module.decoder.encoder.layers.1.fc2.weight, pixel_level_module.decoder.encoder.layers.1.fc2.bias, pixel_level_module.decoder.encoder.layers.1.final_layer_norm.weight, pixel_level_module.decoder.encoder.layers.1.final_layer_norm.bias, pixel_level_module.decoder.encoder.layers.2.self_attn.sampling_offsets.weight, pixel_level_module.decoder.encoder.layers.2.self_attn.sampling_offsets.bias, pixel_level_module.decoder.encoder.layers.2.self_attn.attention_weights.weight, pixel_level_module.decoder.encoder.layers.2.self_attn.attention_weights.bias, pixel_level_module.decoder.encoder.layers.2.self_attn.value_proj.weight, pixel_level_module.decoder.encoder.layers.2.self_attn.value_proj.bias, pixel_level_module.decoder.encoder.layers.2.self_attn.output_proj.weight, pixel_level_module.decoder.encoder.layers.2.self_attn.output_proj.bias, pixel_level_module.decoder.encoder.layers.2.self_attn_layer_norm.weight, pixel_level_module.decoder.encoder.layers.2.self_attn_layer_norm.bias, pixel_level_module.decoder.encoder.layers.2.fc1.weight, pixel_level_module.decoder.encoder.layers.2.fc1.bias, pixel_level_module.decoder.encoder.layers.2.fc2.weight, pixel_level_module.decoder.encoder.layers.2.fc2.bias, pixel_level_module.decoder.encoder.layers.2.final_layer_norm.weight, pixel_level_module.decoder.encoder.layers.2.final_layer_norm.bias, pixel_level_module.decoder.encoder.layers.3.self_attn.sampling_offsets.weight, pixel_level_module.decoder.encoder.layers.3.self_attn.sampling_offsets.bias, pixel_level_module.decoder.encoder.layers.3.self_attn.attention_weights.weight, pixel_level_module.decoder.encoder.layers.3.self_attn.attention_weights.bias, pixel_level_module.decoder.encoder.layers.3.self_attn.value_proj.weight, pixel_level_module.decoder.encoder.layers.3.self_attn.value_proj.bias, pixel_level_module.decoder.encoder.layers.3.self_attn.output_proj.weight, pixel_level_module.decoder.encoder.layers.3.self_attn.output_proj.bias, pixel_level_module.decoder.encoder.layers.3.self_attn_layer_norm.weight, pixel_level_module.decoder.encoder.layers.3.self_attn_layer_norm.bias, pixel_level_module.decoder.encoder.layers.3.fc1.weight, pixel_level_module.decoder.encoder.layers.3.fc1.bias, pixel_level_module.decoder.encoder.layers.3.fc2.weight, pixel_level_module.decoder.encoder.layers.3.fc2.bias, pixel_level_module.decoder.encoder.layers.3.final_layer_norm.weight, pixel_level_module.decoder.encoder.layers.3.final_layer_norm.bias, pixel_level_module.decoder.encoder.layers.4.self_attn.sampling_offsets.weight, pixel_level_module.decoder.encoder.layers.4.self_attn.sampling_offsets.bias, pixel_level_module.decoder.encoder.layers.4.self_attn.attention_weights.weight, pixel_level_module.decoder.encoder.layers.4.self_attn.attention_weights.bias, pixel_level_module.decoder.encoder.layers.4.self_attn.value_proj.weight, pixel_level_module.decoder.encoder.layers.4.self_attn.value_proj.bias, pixel_level_module.decoder.encoder.layers.4.self_attn.output_proj.weight, pixel_level_module.decoder.encoder.layers.4.self_attn.output_proj.bias, pixel_level_module.decoder.encoder.layers.4.self_attn_layer_norm.weight, pixel_level_module.decoder.encoder.layers.4.self_attn_layer_norm.bias, pixel_level_module.decoder.encoder.layers.4.fc1.weight, pixel_level_module.decoder.encoder.layers.4.fc1.bias, pixel_level_module.decoder.encoder.layers.4.fc2.weight, pixel_level_module.decoder.encoder.layers.4.fc2.bias, pixel_level_module.decoder.encoder.layers.4.final_layer_norm.weight, pixel_level_module.decoder.encoder.layers.4.final_layer_norm.bias, pixel_level_module.decoder.encoder.layers.5.self_attn.sampling_offsets.weight, pixel_level_module.decoder.encoder.layers.5.self_attn.sampling_offsets.bias, pixel_level_module.decoder.encoder.layers.5.self_attn.attention_weights.weight, pixel_level_module.decoder.encoder.layers.5.self_attn.attention_weights.bias, pixel_level_module.decoder.encoder.layers.5.self_attn.value_proj.weight, pixel_level_module.decoder.encoder.layers.5.self_attn.value_proj.bias, pixel_level_module.decoder.encoder.layers.5.self_attn.output_proj.weight, pixel_level_module.decoder.encoder.layers.5.self_attn.output_proj.bias, pixel_level_module.decoder.encoder.layers.5.self_attn_layer_norm.weight, pixel_level_module.decoder.encoder.layers.5.self_attn_layer_norm.bias, pixel_level_module.decoder.encoder.layers.5.fc1.weight, pixel_level_module.decoder.encoder.layers.5.fc1.bias, pixel_level_module.decoder.encoder.layers.5.fc2.weight, pixel_level_module.decoder.encoder.layers.5.fc2.bias, pixel_level_module.decoder.encoder.layers.5.final_layer_norm.weight, pixel_level_module.decoder.encoder.layers.5.final_layer_norm.bias, pixel_level_module.decoder.mask_projection.weight, pixel_level_module.decoder.mask_projection.bias, pixel_level_module.decoder.adapter_1.0.weight, pixel_level_module.decoder.adapter_1.1.weight, pixel_level_module.decoder.adapter_1.1.bias, pixel_level_module.decoder.layer_1.0.weight, pixel_level_module.decoder.layer_1.1.weight, pixel_level_module.decoder.layer_1.1.bias, transformer_module.queries_embedder.weight, transformer_module.queries_features.weight, transformer_module.decoder.layers.0.self_attn.k_proj.weight, transformer_module.decoder.layers.0.self_attn.k_proj.bias, transformer_module.decoder.layers.0.self_attn.v_proj.weight, transformer_module.decoder.layers.0.self_attn.v_proj.bias, transformer_module.decoder.layers.0.self_attn.q_proj.weight, transformer_module.decoder.layers.0.self_attn.q_proj.bias, transformer_module.decoder.layers.0.self_attn.out_proj.weight, transformer_module.decoder.layers.0.self_attn.out_proj.bias, transformer_module.decoder.layers.0.self_attn_layer_norm.weight, transformer_module.decoder.layers.0.self_attn_layer_norm.bias, transformer_module.decoder.layers.0.cross_attn.in_proj_weight, transformer_module.decoder.layers.0.cross_attn.in_proj_bias, transformer_module.decoder.layers.0.cross_attn.out_proj.weight, transformer_module.decoder.layers.0.cross_attn.out_proj.bias, transformer_module.decoder.layers.0.cross_attn_layer_norm.weight, transformer_module.decoder.layers.0.cross_attn_layer_norm.bias, transformer_module.decoder.layers.0.fc1.weight, transformer_module.decoder.layers.0.fc1.bias, transformer_module.decoder.layers.0.fc2.weight, transformer_module.decoder.layers.0.fc2.bias, transformer_module.decoder.layers.0.final_layer_norm.weight, transformer_module.decoder.layers.0.final_layer_norm.bias, transformer_module.decoder.layers.1.self_attn.k_proj.weight, transformer_module.decoder.layers.1.self_attn.k_proj.bias, transformer_module.decoder.layers.1.self_attn.v_proj.weight, transformer_module.decoder.layers.1.self_attn.v_proj.bias, transformer_module.decoder.layers.1.self_attn.q_proj.weight, transformer_module.decoder.layers.1.self_attn.q_proj.bias, transformer_module.decoder.layers.4.self_attn.k_proj.weight, transformer_module.decoder.layers.4.self_attn.k_proj.bias, transformer_module.decoder.layers.4.self_attn.v_proj.weight, transformer_module.decoder.layers.4.self_attn.v_proj.bias, transformer_module.decoder.layers.4.self_attn.q_proj.weight, transformer_module.decoder.layers.4.self_attn.q_proj.bias, transformer_module.decoder.layers.4.self_attn.out_proj.weight, transformer_module.decoder.layers.4.self_attn.out_proj.bias, transformer_module.decoder.layers.4.self_attn_layer_norm.weight, transformer_module.decoder.layers.4.self_attn_layer_norm.bias, transformer_module.decoder.layers.4.cross_attn.in_proj_weight, transformer_module.decoder.layers.4.cross_attn.in_proj_bias, transformer_module.decoder.layers.4.cross_attn.out_proj.weight, transformer_module.decoder.layers.4.cross_attn.out_proj.bias, transformer_module.decoder.layers.4.cross_attn_layer_norm.weight, transformer_module.decoder.layers.4.cross_attn_layer_norm.bias, transformer_module.decoder.layers.4.fc1.weight, transformer_module.decoder.layers.4.fc1.bias, transformer_module.decoder.layers.4.fc2.weight, transformer_module.decoder.layers.4.fc2.bias, transformer_module.decoder.layers.4.final_layer_norm.weight, transformer_module.decoder.layers.4.final_layer_norm.bias, transformer_module.decoder.layers.5.self_attn.k_proj.weight, transformer_module.decoder.layers.5.self_attn.k_proj.bias, transformer_module.decoder.layers.5.self_attn.v_proj.weight, transformer_module.decoder.layers.5.self_attn.v_proj.bias, transformer_module.decoder.layers.5.self_attn.q_proj.weight, transformer_module.decoder.layers.5.self_attn.q_proj.bias, transformer_module.decoder.layers.5.self_attn.out_proj.weight, transformer_module.decoder.layers.5.self_attn.out_proj.bias, transformer_module.decoder.layers.5.self_attn_layer_norm.weight, transformer_module.decoder.layers.5.self_attn_layer_norm.bias, transformer_module.decoder.layers.5.cross_attn.in_proj_weight, transformer_module.decoder.layers.5.cross_attn.in_proj_bias, transformer_module.decoder.layers.5.cross_attn.out_proj.weight, transformer_module.decoder.layers.5.cross_attn.out_proj.bias, transformer_module.decoder.layers.5.cross_attn_layer_norm.weight, transformer_module.decoder.layers.5.cross_attn_layer_norm.bias, transformer_module.decoder.layers.5.fc1.weight, transformer_module.decoder.layers.5.fc1.bias, transformer_module.decoder.layers.5.fc2.weight, transformer_module.decoder.layers.5.fc2.bias, transformer_module.decoder.layers.5.final_layer_norm.weight, transformer_module.decoder.layers.5.final_layer_norm.bias, transformer_module.decoder.layers.6.self_attn.k_proj.weight, transformer_module.decoder.layers.6.self_attn.k_proj.bias, transformer_module.decoder.layers.6.self_attn.v_proj.weight, transformer_module.decoder.layers.6.self_attn.v_proj.bias, transformer_module.decoder.layers.6.self_attn.q_proj.weight, transformer_module.decoder.layers.6.self_attn.q_proj.bias, transformer_module.decoder.layers.6.self_attn.out_proj.weight, transformer_module.decoder.layers.6.self_attn.out_proj.bias, transformer_module.decoder.layers.6.self_attn_layer_norm.weight, transformer_module.decoder.layers.6.self_attn_layer_norm.bias, transformer_module.decoder.layers.6.cross_attn.in_proj_weight, transformer_module.decoder.layers.6.cross_attn.in_proj_bias, transformer_module.decoder.layers.6.cross_attn.out_proj.weight, transformer_module.decoder.layers.6.cross_attn.out_proj.bias, transformer_module.decoder.layers.6.cross_attn_layer_norm.weight, transformer_module.decoder.layers.6.cross_attn_layer_norm.bias, transformer_module.decoder.layers.6.fc1.weight, transformer_module.decoder.layers.6.fc1.bias, transformer_module.decoder.layers.6.fc2.weight, transformer_module.decoder.layers.6.fc2.bias, transformer_module.decoder.layers.6.final_layer_norm.weight, transformer_module.decoder.layers.6.final_layer_norm.bias, transformer_module.decoder.layers.7.self_attn.k_proj.weight, transformer_module.decoder.layers.7.self_attn.k_proj.bias, transformer_module.decoder.layers.7.self_attn.v_proj.weight, transformer_module.decoder.layers.7.self_attn.v_proj.bias, transformer_module.decoder.layers.7.self_attn.q_proj.weight, transformer_module.decoder.layers.7.self_attn.q_proj.bias, transformer_module.decoder.layers.7.self_attn.out_proj.weight, transformer_module.decoder.layers.7.self_attn.out_proj.bias, transformer_module.decoder.layers.7.self_attn_layer_norm.weight, transformer_module.decoder.layers.7.self_attn_layer_norm.bias, transformer_module.decoder.layers.7.cross_attn.in_proj_weight, transformer_module.decoder.layers.7.cross_attn.in_proj_bias, transformer_module.decoder.layers.7.cross_attn.out_proj.weight, transformer_module.decoder.layers.7.cross_attn.out_proj.bias, transformer_module.decoder.layers.7.cross_attn_layer_norm.weight, transformer_module.decoder.layers.7.cross_attn_layer_norm.bias, transformer_module.decoder.layers.7.fc1.weight, transformer_module.decoder.layers.7.fc1.bias, transformer_module.decoder.layers.7.fc2.weight, transformer_module.decoder.layers.7.fc2.bias, transformer_module.decoder.layers.7.final_layer_norm.weight, transformer_module.decoder.layers.7.final_layer_norm.bias, transformer_module.decoder.layers.8.self_attn.k_proj.weight, transformer_module.decoder.layers.8.self_attn.k_proj.bias, transformer_module.decoder.layers.8.self_attn.v_proj.weight, transformer_module.decoder.layers.8.self_attn.v_proj.bias, transformer_module.decoder.layers.8.self_attn.q_proj.weight, transformer_module.decoder.layers.8.self_attn.q_proj.bias, transformer_module.decoder.layers.8.self_attn.out_proj.weight, transformer_module.decoder.layers.8.self_attn.out_proj.bias, transformer_module.decoder.layers.8.self_attn_layer_norm.weight, transformer_module.decoder.layers.8.self_attn_layer_norm.bias, transformer_module.decoder.layers.8.cross_attn.in_proj_weight, transformer_module.decoder.layers.8.cross_attn.in_proj_bias, transformer_module.decoder.layers.8.cross_attn.out_proj.weight, transformer_module.decoder.layers.8.cross_attn.out_proj.bias, transformer_module.decoder.layers.8.cross_attn_layer_norm.weight, transformer_module.decoder.layers.8.cross_attn_layer_norm.bias, transformer_module.decoder.layers.8.fc1.weight, transformer_module.decoder.layers.8.fc1.bias, transformer_module.decoder.layers.8.fc2.weight, transformer_module.decoder.layers.8.fc2.bias, transformer_module.decoder.layers.8.final_layer_norm.weight, transformer_module.decoder.layers.8.final_layer_norm.bias, transformer_module.decoder.layernorm.weight, transformer_module.decoder.layernorm.bias, transformer_module.decoder.mask_predictor.mask_embedder.0.0.weight, transformer_module.decoder.mask_predictor.mask_embedder.0.0.bias, transformer_module.decoder.mask_predictor.mask_embedder.1.0.weight, transformer_module.decoder.mask_predictor.mask_embedder.1.0.bias, transformer_module.decoder.mask_predictor.mask_embedder.2.0.weight, transformer_module.decoder.mask_predictor.mask_embedder.2.0.bias, transformer_module.level_embed.weight, class_predictor.weight, class_predictor.bias, criterion.empty_weight
missing keys in source state_dict: backbone.patch_embed.projection.weight, backbone.patch_embed.projection.bias, backbone.patch_embed.norm.weight, backbone.patch_embed.norm.bias, backbone.stages.0.blocks.0.norm1.weight, backbone.stages.0.blocks.0.norm1.bias, backbone.stages.0.blocks.0.attn.w_msa.relative_position_bias_table, backbone.stages.0.blocks.0.attn.w_msa.relative_position_index, backbone.stages.0.blocks.0.attn.w_msa.qkv.weight, backbone.stages.0.blocks.0.attn.w_msa.qkv.bias, backbone.stages.0.blocks.0.attn.w_msa.proj.weight, backbone.stages.0.blocks.0.attn.w_msa.proj.bias, backbone.stages.0.blocks.0.norm2.weight, backbone.stages.0.blocks.0.norm2.bias, backbone.stages.0.blocks.0.ffn.layers.0.0.weight, backbone.stages.0.blocks.0.ffn.layers.0.0.bias, backbone.stages.0.blocks.0.ffn.layers.1.weight, backbone.stages.0.blocks.0.ffn.layers.1.bias, backbone.stages.0.blocks.1.norm1.weight, backbone.stages.0.blocks.1.norm1.bias, backbone.stages.0.blocks.1.attn.w_msa.relative_position_bias_table, backbone.stages.0.blocks.1.attn.w_msa.relative_position_index, backbone.stages.0.blocks.1.attn.w_msa.qkv.weight, backbone.stages.0.blocks.1.attn.w_msa.qkv.bias, backbone.stages.0.blocks.1.attn.w_msa.proj.weight, backbone.stages.0.blocks.1.attn.w_msa.proj.bias, backbone.stages.0.blocks.1.norm2.weight, backbone.stages.0.blocks.1.norm2.bias, backbone.stages.0.blocks.1.ffn.layers.0.0.weight, backbone.stages.0.blocks.1.ffn.layers.0.0.bias, backbone.stages.0.blocks.1.ffn.layers.1.weight, backbone.stages.0.blocks.1.ffn.layers.1.bias, backbone.stages.0.downsample.norm.weight, backbone.stages.0.downsample.norm.bias, backbone.stages.0.downsample.reduction.weight, backbone.stages.1.blocks.0.norm1.weight, backbone.stages.1.blocks.0.norm1.bias, backbone.stages.1.blocks.0.attn.w_msa.relative_position_bias_table, backbone.stages.1.blocks.0.attn.w_msa.relative_position_index, backbone.stages.1.blocks.0.attn.w_msa.qkv.weight, backbone.stages.1.blocks.0.attn.w_msa.qkv.bias, backbone.stages.1.blocks.0.attn.w_msa.proj.weight, backbone.stages.1.blocks.0.attn.w_msa.proj.bias, backbone.stages.1.blocks.0.norm2.weight, backbone.stages.1.blocks.0.norm2.bias, backbone.stages.1.blocks.0.ffn.layers.0.0.weight, backbone.stages.1.blocks.0.ffn.layers.0.0.bias, backbone.stages.1.blocks.0.ffn.layers.1.weight, backbone.stages.1.blocks.0.ffn.layers.1.bias, backbone.stages.1.blocks.1.norm1.weight, backbone.stages.1.blocks.1.norm1.bias, backbone.stages.1.blocks.1.attn.w_msa.relative_position_bias_table, backbone.stages.1.blocks.1.attn.w_msa.relative_position_index, backbone.stages.1.blocks.1.attn.w_msa.qkv.weight, backbone.stages.1.blocks.1.attn.w_msa.qkv.bias, backbone.stages.1.blocks.1.attn.w_msa.proj.weight, backbone.stages.1.blocks.1.attn.w_msa.proj.bias, backbone.stages.1.blocks.1.norm2.weight, backbone.stages.1.blocks.1.norm2.bias, backbone.stages.1.blocks.1.ffn.layers.0.0.weight, backbone.stages.1.blocks.1.ffn.layers.0.0.bias, backbone.stages.1.blocks.1.ffn.layers.1.weight, backbone.stages.1.blocks.1.ffn.layers.1.bias, backbone.stages.1.downsample.norm.weight, backbone.stages.1.downsample.norm.bias, backbone.stages.1.downsample.reduction.weight, backbone.stages.2.blocks.0.norm1.weight, backbone.stages.2.blocks.0.norm1.bias, backbone.stages.2.blocks.0.attn.w_msa.relative_position_bias_table, backbone.stages.2.blocks.0.attn.w_msa.relative_position_index, backbone.stages.2.blocks.0.attn.w_msa.qkv.weight, backbone.stages.2.blocks.0.attn.w_msa.qkv.bias, backbone.stages.2.blocks.0.attn.w_msa.proj.weight, backbone.stages.2.blocks.0.attn.w_msa.proj.bias, backbone.stages.2.blocks.0.norm2.weight, backbone.stages.2.blocks.0.norm2.bias, backbone.stages.2.blocks.0.ffn.layers.0.0.weight, backbone.stages.2.blocks.0.ffn.layers.0.0.bias, backbone.stages.2.blocks.0.ffn.layers.1.weight, backbone.stages.2.blocks.0.ffn.layers.1.bias, backbone.stages.2.blocks.1.norm1.weight, backbone.stages.2.blocks.1.norm1.bias, backbone.stages.2.blocks.1.attn.w_msa.relative_position_bias_table, backbone.stages.2.blocks.1.attn.w_msa.relative_position_index, backbone.stages.2.blocks.1.attn.w_msa.qkv.weight, backbone.stages.2.blocks.1.attn.w_msa.qkv.bias, backbone.stages.2.blocks.1.attn.w_msa.proj.weight, backbone.stages.2.blocks.1.attn.w_msa.proj.bias, backbone.stages.2.blocks.1.norm2.weight, backbone.stages.2.blocks.1.norm2.bias, backbone.stages.2.blocks.1.ffn.layers.0.0.weight, backbone.stages.2.blocks.1.ffn.layers.0.0.bias, backbone.stages.2.blocks.1.ffn.layers.1.weight, backbone.stages.2.blocks.1.ffn.layers.1.bias, backbone.stages.2.blocks.2.norm1.weight, backbone.stages.2.blocks.2.norm1.bias, backbone.stages.2.blocks.2.attn.w_msa.relative_position_bias_table, backbone.stages.2.blocks.2.attn.w_msa.relative_position_index, backbone.stages.2.blocks.2.attn.w_msa.qkv.weight, backbone.stages.2.blocks.2.attn.w_msa.qkv.bias, backbone.stages.2.blocks.2.attn.w_msa.proj.weight, backbone.stages.2.blocks.2.attn.w_msa.proj.bias, backbone.stages.3.blocks.1.attn.w_msa.qkv.weight, backbone.stages.3.blocks.1.attn.w_msa.qkv.bias, backbone.stages.3.blocks.1.attn.w_msa.proj.weight, backbone.stages.3.blocks.1.attn.w_msa.proj.bias, backbone.stages.3.blocks.1.norm2.weight, backbone.stages.3.blocks.1.norm2.bias, backbone.stages.3.blocks.1.ffn.layers.0.0.weight, backbone.stages.3.blocks.1.ffn.layers.0.0.bias, backbone.stages.3.blocks.1.ffn.layers.1.weight, backbone.stages.3.blocks.1.ffn.layers.1.bias, backbone.norm0.weight, backbone.norm0.bias, backbone.norm1.weight, backbone.norm1.bias, backbone.norm2.weight, backbone.norm2.bias, backbone.norm3.weight, backbone.norm3.bias, decode_head.pixel_decoder.input_convs.0.conv.weight, decode_head.pixel_decoder.input_convs.0.conv.bias, decode_head.pixel_decoder.input_convs.0.gn.weight, decode_head.pixel_decoder.input_convs.0.gn.bias, decode_head.pixel_decoder.input_convs.1.conv.weight, decode_head.pixel_decoder.input_convs.1.conv.bias, decode_head.pixel_decoder.input_convs.1.gn.weight, decode_head.pixel_decoder.input_convs.1.gn.bias, decode_head.pixel_decoder.input_convs.2.conv.weight, decode_head.pixel_decoder.input_convs.2.conv.bias, decode_head.pixel_decoder.input_convs.2.gn.weight, decode_head.pixel_decoder.input_convs.2.gn.bias, decode_head.pixel_decoder.encoder.layers.0.self_attn.sampling_offsets.weight, decode_head.pixel_decoder.encoder.layers.0.self_attn.sampling_offsets.bias, decode_head.pixel_decoder.encoder.layers.0.self_attn.attention_weights.weight, decode_head.pixel_decoder.encoder.layers.0.self_attn.attention_weights.bias, decode_head.pixel_decoder.encoder.layers.0.self_attn.value_proj.weight, decode_head.pixel_decoder.encoder.layers.0.self_attn.value_proj.bias, decode_head.pixel_decoder.encoder.layers.0.self_attn.output_proj.weight, decode_head.pixel_decoder.encoder.layers.0.self_attn.output_proj.bias, decode_head.pixel_decoder.encoder.layers.0.ffn.layers.0.0.weight, decode_head.pixel_decoder.encoder.layers.0.ffn.layers.0.0.bias, decode_head.pixel_decoder.encoder.layers.0.ffn.layers.1.weight, decode_head.pixel_decoder.encoder.layers.0.ffn.layers.1.bias, decode_head.pixel_decoder.encoder.layers.0.norms.0.weight, decode_head.pixel_decoder.encoder.layers.0.norms.0.bias, decode_head.pixel_decoder.encoder.layers.0.norms.1.weight, decode_head.pixel_decoder.encoder.layers.0.norms.1.bias, decode_head.pixel_decoder.encoder.layers.1.self_attn.sampling_offsets.weight, decode_head.pixel_decoder.encoder.layers.1.self_attn.sampling_offsets.bias, decode_head.pixel_decoder.encoder.layers.1.self_attn.attention_weights.weight, decode_head.pixel_decoder.encoder.layers.1.self_attn.attention_weights.bias, decode_head.pixel_decoder.encoder.layers.1.self_attn.value_proj.weight, decode_head.pixel_decoder.encoder.layers.1.self_attn.value_proj.bias, decode_head.pixel_decoder.encoder.layers.1.self_attn.output_proj.weight, decode_head.pixel_decoder.encoder.layers.1.self_attn.output_proj.bias, decode_head.pixel_decoder.encoder.layers.1.ffn.layers.0.0.weight, decode_head.pixel_decoder.encoder.layers.1.ffn.layers.0.0.bias, decode_head.pixel_decoder.encoder.layers.1.ffn.layers.1.weight, decode_head.pixel_decoder.encoder.layers.1.ffn.layers.1.bias, decode_head.pixel_decoder.encoder.layers.1.norms.0.weight, decode_head.pixel_decoder.encoder.layers.1.norms.0.bias, decode_head.pixel_decoder.encoder.layers.1.norms.1.weight, decode_head.pixel_decoder.encoder.layers.1.norms.1.bias, decode_head.pixel_decoder.encoder.layers.2.self_attn.sampling_offsets.weight, decode_head.pixel_decoder.encoder.layers.2.self_attn.sampling_offsets.bias, decode_head.pixel_decoder.encoder.layers.2.self_attn.attention_weights.weight, decode_head.pixel_decoder.encoder.layers.2.self_attn.attention_weights.bias, decode_head.pixel_decoder.encoder.layers.2.self_attn.value_proj.weight, decode_head.pixel_decoder.encoder.layers.2.self_attn.value_proj.bias, decode_head.pixel_decoder.encoder.layers.2.self_attn.output_proj.weight, decode_head.pixel_decoder.encoder.layers.2.self_attn.output_proj.bias, decode_head.pixel_decoder.encoder.layers.2.ffn.layers.0.0.weight, decode_head.pixel_decoder.encoder.layers.2.ffn.layers.0.0.bias, decode_head.pixel_decoder.encoder.layers.2.ffn.layers.1.weight, decode_head.pixel_decoder.encoder.layers.2.ffn.layers.1.bias, decode_head.pixel_decoder.encoder.layers.2.norms.0.weight, decode_head.pixel_decoder.encoder.layers.2.norms.0.bias, decode_head.pixel_decoder.encoder.layers.2.norms.1.weight, decode_head.pixel_decoder.encoder.layers.2.norms.1.bias, decode_head.pixel_decoder.encoder.layers.3.self_attn.sampling_offsets.weight, decode_head.pixel_decoder.encoder.layers.3.self_attn.sampling_offsets.bias, decode_head.pixel_decoder.encoder.layers.3.self_attn.attention_weights.weight, decode_head.pixel_decoder.encoder.layers.3.self_attn.attention_weights.bias, decode_head.pixel_decoder.encoder.layers.3.self_attn.value_proj.weight, decode_head.pixel_decoder.encoder.layers.3.self_attn.value_proj.bias, decode_head.pixel_decoder.encoder.layers.3.self_attn.output_proj.weight, decode_head.pixel_decoder.encoder.layers.3.self_attn.output_proj.bias, decode_head.pixel_decoder.encoder.layers.3.ffn.layers.0.0.weight, decode_head.pixel_decoder.encoder.layers.3.ffn.layers.0.0.bias, decode_head.pixel_decoder.encoder.layers.3.ffn.layers.1.weight, decode_head.pixel_decoder.encoder.layers.3.ffn.layers.1.bias, decode_head.pixel_decoder.encoder.layers.3.norms.0.weight, decode_head.pixel_decoder.encoder.layers.3.norms.0.bias, decode_head.pixel_decoder.encoder.layers.3.norms.1.weight, decode_head.pixel_decoder.encoder.layers.3.norms.1.bias, decode_head.pixel_decoder.encoder.layers.4.self_attn.sampling_offsets.weight, decode_head.pixel_decoder.encoder.layers.4.self_attn.sampling_offsets.bias, decode_head.pixel_decoder.encoder.layers.4.self_attn.attention_weights.weight, decode_head.pixel_decoder.encoder.layers.4.self_attn.attention_weights.bias, decode_head.pixel_decoder.encoder.layers.4.self_attn.value_proj.weight, decode_head.pixel_decoder.encoder.layers.4.self_attn.value_proj.bias, decode_head.pixel_decoder.encoder.layers.4.self_attn.output_proj.weight, decode_head.pixel_decoder.encoder.layers.4.self_attn.output_proj.bias, decode_head.pixel_decoder.encoder.layers.4.ffn.layers.0.0.weight, decode_head.pixel_decoder.encoder.layers.4.ffn.layers.0.0.bias, decode_head.pixel_decoder.encoder.layers.4.ffn.layers.1.weight, decode_head.pixel_decoder.encoder.layers.4.ffn.layers.1.bias, decode_head.pixel_decoder.encoder.layers.4.norms.0.weight, decode_head.pixel_decoder.encoder.layers.4.norms.0.bias, decode_head.pixel_decoder.encoder.layers.4.norms.1.weight, decode_head.pixel_decoder.encoder.layers.4.norms.1.bias, decode_head.pixel_decoder.encoder.layers.5.self_attn.sampling_offsets.weight, decode_head.pixel_decoder.encoder.layers.5.self_attn.sampling_offsets.bias, decode_head.pixel_decoder.encoder.layers.5.self_attn.attention_weights.weight, decode_head.pixel_decoder.encoder.layers.5.self_attn.attention_weights.bias, decode_head.pixel_decoder.encoder.layers.5.self_attn.value_proj.weight, decode_head.pixel_decoder.encoder.layers.5.self_attn.value_proj.bias, decode_head.pixel_decoder.encoder.layers.5.self_attn.output_proj.weight, decode_head.pixel_decoder.encoder.layers.5.self_attn.output_proj.bias, decode_head.pixel_decoder.encoder.layers.5.ffn.layers.0.0.weight, decode_head.pixel_decoder.encoder.layers.5.ffn.layers.0.0.bias, decode_head.pixel_decoder.encoder.layers.5.ffn.layers.1.weight, decode_head.pixel_decoder.encoder.layers.5.ffn.layers.1.bias, decode_head.pixel_decoder.encoder.layers.5.norms.0.weight, decode_head.pixel_decoder.encoder.layers.5.norms.0.bias, decode_head.pixel_decoder.encoder.layers.5.norms.1.weight, decode_head.pixel_decoder.encoder.layers.5.norms.1.bias, decode_head.pixel_decoder.level_encoding.weight, decode_head.pixel_decoder.lateral_convs.0.conv.weight, decode_head.pixel_decoder.lateral_convs.0.gn.weight, decode_head.pixel_decoder.lateral_convs.0.gn.bias, decode_head.pixel_decoder.output_convs.0.conv.weight, decode_head.pixel_decoder.output_convs.0.gn.weight, decode_head.pixel_decoder.output_convs.0.gn.bias, decode_head.pixel_decoder.mask_feature.weight, decode_head.pixel_decoder.mask_feature.bias, decode_head.transformer_decoder.layers.0.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.0.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.0.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.0.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.0.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.0.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.0.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.0.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.0.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.0.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.0.ffn.layers.1.weight, decode_head.transformer_decoder.layers.0.ffn.layers.1.bias, decode_head.transformer_decoder.layers.0.norms.0.weight, decode_head.transformer_decoder.layers.0.norms.0.bias, decode_head.transformer_decoder.layers.0.norms.1.weight, decode_head.transformer_decoder.layers.0.norms.1.bias, decode_head.transformer_decoder.layers.0.norms.2.weight, decode_head.transformer_decoder.layers.0.norms.2.bias, decode_head.transformer_decoder.layers.1.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.1.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.1.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.1.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.1.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.1.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.1.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.1.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.1.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.1.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.1.ffn.layers.1.weight, decode_head.transformer_decoder.layers.1.ffn.layers.1.bias, decode_head.transformer_decoder.layers.1.norms.0.weight, decode_head.transformer_decoder.layers.1.norms.0.bias, decode_head.transformer_decoder.layers.1.norms.1.weight, decode_head.transformer_decoder.layers.1.norms.1.bias, decode_head.transformer_decoder.layers.1.norms.2.weight, decode_head.transformer_decoder.layers.1.norms.2.bias, decode_head.transformer_decoder.layers.2.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.2.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.2.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.2.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.2.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.2.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.2.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.2.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.2.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.2.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.2.ffn.layers.1.weight, decode_head.transformer_decoder.layers.2.ffn.layers.1.bias, decode_head.transformer_decoder.layers.2.norms.0.weight, decode_head.transformer_decoder.layers.2.norms.0.bias, decode_head.transformer_decoder.layers.2.norms.1.weight, decode_head.transformer_decoder.layers.2.norms.1.bias, decode_head.transformer_decoder.layers.2.norms.2.weight, decode_head.transformer_decoder.layers.2.norms.2.bias, decode_head.transformer_decoder.layers.3.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.3.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.3.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.3.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.3.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.3.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.3.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.3.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.3.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.3.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.3.ffn.layers.1.weight, decode_head.transformer_decoder.layers.3.ffn.layers.1.bias, decode_head.transformer_decoder.layers.3.norms.0.weight, decode_head.transformer_decoder.layers.3.norms.0.bias, decode_head.transformer_decoder.layers.3.norms.1.weight, decode_head.transformer_decoder.layers.3.norms.1.bias, decode_head.transformer_decoder.layers.3.norms.2.weight, decode_head.transformer_decoder.layers.3.norms.2.bias, decode_head.transformer_decoder.layers.4.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.4.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.4.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.4.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.4.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.4.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.4.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.4.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.4.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.4.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.4.ffn.layers.1.weight, decode_head.transformer_decoder.layers.4.ffn.layers.1.bias, decode_head.transformer_decoder.layers.4.norms.0.weight, decode_head.transformer_decoder.layers.4.norms.0.bias, decode_head.transformer_decoder.layers.4.norms.1.weight, decode_head.transformer_decoder.layers.4.norms.1.bias, decode_head.transformer_decoder.layers.4.norms.2.weight, decode_head.transformer_decoder.layers.4.norms.2.bias, decode_head.transformer_decoder.layers.5.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.5.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.5.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.5.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.5.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.5.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.5.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.5.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.5.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.5.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.5.ffn.layers.1.weight, decode_head.transformer_decoder.layers.5.ffn.layers.1.bias, decode_head.transformer_decoder.layers.5.norms.0.weight, decode_head.transformer_decoder.layers.5.norms.0.bias, decode_head.transformer_decoder.layers.5.norms.1.weight, decode_head.transformer_decoder.layers.5.norms.1.bias, decode_head.transformer_decoder.layers.5.norms.2.weight, decode_head.transformer_decoder.layers.5.norms.2.bias, decode_head.transformer_decoder.layers.6.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.6.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.6.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.6.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.6.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.6.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.6.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.6.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.6.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.6.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.6.ffn.layers.1.weight, decode_head.transformer_decoder.layers.6.ffn.layers.1.bias, decode_head.transformer_decoder.layers.6.norms.0.weight, decode_head.transformer_decoder.layers.6.norms.0.bias, decode_head.transformer_decoder.layers.6.norms.1.weight, decode_head.transformer_decoder.layers.6.norms.1.bias, decode_head.transformer_decoder.layers.6.norms.2.weight, decode_head.transformer_decoder.layers.6.norms.2.bias, decode_head.transformer_decoder.layers.7.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.7.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.7.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.7.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.7.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.7.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.7.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.7.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.7.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.7.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.7.ffn.layers.1.weight, decode_head.transformer_decoder.layers.7.ffn.layers.1.bias, decode_head.transformer_decoder.layers.7.norms.0.weight, decode_head.transformer_decoder.layers.7.norms.0.bias, decode_head.transformer_decoder.layers.7.norms.1.weight, decode_head.transformer_decoder.layers.7.norms.1.bias, decode_head.transformer_decoder.layers.7.norms.2.weight, decode_head.transformer_decoder.layers.7.norms.2.bias, decode_head.transformer_decoder.layers.8.self_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.8.self_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.8.self_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.8.self_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.8.cross_attn.attn.in_proj_weight, decode_head.transformer_decoder.layers.8.cross_attn.attn.in_proj_bias, decode_head.transformer_decoder.layers.8.cross_attn.attn.out_proj.weight, decode_head.transformer_decoder.layers.8.cross_attn.attn.out_proj.bias, decode_head.transformer_decoder.layers.8.ffn.layers.0.0.weight, decode_head.transformer_decoder.layers.8.ffn.layers.0.0.bias, decode_head.transformer_decoder.layers.8.ffn.layers.1.weight, decode_head.transformer_decoder.layers.8.ffn.layers.1.bias, decode_head.transformer_decoder.layers.8.norms.0.weight, decode_head.transformer_decoder.layers.8.norms.0.bias, decode_head.transformer_decoder.layers.8.norms.1.weight, decode_head.transformer_decoder.layers.8.norms.1.bias, decode_head.transformer_decoder.layers.8.norms.2.weight, decode_head.transformer_decoder.layers.8.norms.2.bias, decode_head.transformer_decoder.post_norm.weight, decode_head.transformer_decoder.post_norm.bias, decode_head.query_embed.weight, decode_head.query_feat.weight, decode_head.level_embed.weight, decode_head.cls_embed.weight, decode_head.cls_embed.bias, decode_head.mask_embed.0.weight, decode_head.mask_embed.0.bias, decode_head.mask_embed.2.weight, decode_head.mask_embed.2.bias, decode_head.mask_embed.4.weight, decode_head.mask_embed.4.bias
File /files/private_data/data_convert/convertONNX/mmsegmentation/mmseg/apis/inference.py:61, in init_model(config, checkpoint, device, cfg_options)
KeyError: 'meta'
- I'm using COCO format dataset, with 50k+ images of building with original size 640x640 and crop size 512x512?
Environment
CUDA available: True
MUSA available: False
numpy_random_seed: 2147483648
GPU 0,1,2,3,4,5,6,7,8,9: NVIDIA A100-PCIE-40GB
CUDA_HOME: /usr/local/cuda
NVCC: Cuda compilation tools, release 12.6, V12.6.77
GCC: gcc (Ubuntu 13.2.0-23ubuntu4) 13.2.0
PyTorch: 2.4.1+cu121
PyTorch compiling details: PyTorch built with:
- GCC 9.3
- C++ Version: 201703
- Intel(R) oneAPI Math Kernel Library Version 2022.2-Product Build 20220804 for Intel(R) 64 architecture applications
- Intel(R) MKL-DNN v3.4.2 (Git Hash 1137e04ec0b5251ca2b4400a4fd3c667ce843d67)
- OpenMP 201511 (a.k.a. OpenMP 4.5)
- LAPACK is enabled (usually provided by MKL)
- NNPACK is enabled
- CPU capability usage: AVX512
- CUDA Runtime 12.1
- NVCC architecture flags: -gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75;-gencode;arch=compute_80,code=sm_80;-gencode;arch=compute_86,code=sm_86;-gencode;arch=compute_90,code=sm_90
- CuDNN 90.1 (built against CUDA 12.4)
- Magma 2.6.1
- Build settings: BLAS_INFO=mkl, BUILD_TYPE=Release, CUDA_VERSION=12.1, CUDNN_VERSION=9.1.0,
TorchVision: 0.12.0
OpenCV: 4.11.0
MMEngine: 0.10.6
MMSegmentation: 1.2.2+b040e14