TUI-NICR / EMSANet

EMSANet: Efficient Multi-Task RGB-D Scene Analysis for Indoor Environments
Apache License 2.0
64 stars 3 forks source link

Evaluation error "RuntimeError: Error(s) in loading state_dict for EMSANet: Missing key(s) in state_dict: " #15

Open AvidahRai opened 1 year ago

AvidahRai commented 1 year ago

RuntimeError: Error(s) in loading state_dict for EMSANet: Missing key(s) in state_dict: "encoder.backbone_rgb.conv1.weight", "encoder.backbone_rgb.norm1.weight", "encoder.backbone_rgb.norm1.bias", "encoder.backbone_rgb.norm1.running_mean", "encoder.backbone_rgb.norm1.running_var", "encoder.backbone_rgb.layer1.0.conv1_1.weight", "encoder.backbone_rgb.layer1.0.conv1_1.bias", "encoder.backbone_rgb.layer1.0.conv1_2.weight", "encoder.backbone_rgb.layer1.0.norm1.weight", "encoder.backbone_rgb.layer1.0.norm1.bias", "encoder.backbone_rgb.layer1.0.norm1.running_mean", "encoder.backbone_rgb.layer1.0.norm1.running_var", "encoder.backbone_rgb.layer1.0.conv2_1.weight", "encoder.backbone_rgb.layer1.0.conv2_1.bias", "encoder.backbone_rgb.layer1.0.conv2_2.weight", "encoder.backbone_rgb.layer1.0.norm2.weight", "encoder.backbone_rgb.layer1.0.norm2.bias", "encoder.backbone_rgb.layer1.0.norm2.running_mean", .........................................................

Tripton commented 1 year ago

Can you please give me the full command you are trying to run to help me identify the issue?