ZoeDepth icon indicating copy to clipboard operation
ZoeDepth copied to clipboard

Unexpected kyes(s) in state_dict

Open chang-xinhai opened this issue 1 year ago • 3 comments

Hi, when I try to use your code with:

import torch
repo = "isl-org/ZoeDepth"
# Zoe_NK
model_zoe_nk = torch.hub.load(repo, "ZoeD_NK", pretrained=True)

This will cause a RuntimeError:

RuntimeError: Error(s) in loading state_dict for ZoeDepthNK:
        Unexpected key(s) in state_dict: "core.core.pretrained.model.blocks.0.attn.relative_position_index", 
"core.core.pretrained.model.blocks.1.attn.relative_position_index", "core.core.pretrained.model.blocks.2.attn.relative_position_index", 
"core.core.pretrained.model.blocks.3.attn.relative_position_index", "core.core.pretrained.model.blocks.4.attn.relative_position_index", 
"core.core.pretrained.model.blocks.5.attn.relative_position_index", "core.core.pretrained.model.blocks.6.attn.relative_position_index", 
"core.core.pretrained.model.blocks.7.attn.relative_position_index", "core.core.pretrained.model.blocks.8.attn.relative_position_index", 
"core.core.pretrained.model.blocks.9.attn.relative_position_index", "core.core.pretrained.model.blocks.10.attn.relative_position_index",
 "core.core.pretrained.model.blocks.11.attn.relative_position_index", "core.core.pretrained.model.blocks.12.attn.relative_position_index", 
"core.core.pretrained.model.blocks.13.attn.relative_position_index", "core.core.pretrained.model.blocks.14.attn.relative_position_index", 
"core.core.pretrained.model.blocks.15.attn.relative_position_index", "core.core.pretrained.model.blocks.16.attn.relative_position_index", 
"core.core.pretrained.model.blocks.17.attn.relative_position_index", "core.core.pretrained.model.blocks.18.attn.relative_position_index", 
"core.core.pretrained.model.blocks.19.attn.relative_position_index", "core.core.pretrained.model.blocks.20.attn.relative_position_index", 
"core.core.pretrained.model.blocks.21.attn.relative_position_index", "core.core.pretrained.model.blocks.22.attn.relative_position_index", 
"core.core.pretrained.model.blocks.23.attn.relative_position_index".

Is there any new version of codes or checkpoint? Thanks.

chang-xinhai avatar Dec 18 '24 02:12 chang-xinhai

same problem

oneway-phil avatar Dec 18 '24 08:12 oneway-phil

Same, too.

AlanLiangC avatar Dec 19 '24 12:12 AlanLiangC

#97 might be helpful.

retazo0018 avatar Dec 24 '24 10:12 retazo0018