Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cannot be evaluated with pretrained models #1

Closed
npduyduc opened this issue Mar 29, 2023 · 1 comment
Closed

Cannot be evaluated with pretrained models #1

npduyduc opened this issue Mar 29, 2023 · 1 comment

Comments

@npduyduc
Copy link

Hi I'm trying to run your eval.sh file with pretrained model and got this error can you help me, thanks

eval.sh
CUDA_VISIBLE_DEVICES=0 python evaluate_depth_HR.py \ --eval_stereo \ --load_weights_folder pretrained/stage1 \ --models_to_load encoder depth \ --use_denseaspp \ --plane_residual \ --width 1280 \ --height 384

error
RuntimeError: Error(s) in loading state_dict for DepthDecoder: Missing key(s) in state_dict: "decoder.13.0.weight", "decoder.13.0.bias", "decoder.13.2.weight", "decoder.13.2.bias". Unexpected key(s) in state_dict: "decoder.14.0.weight", "decoder.14.0.bias", "decoder.14.2.weight", "decoder.14.2.bias", "decoder.13.conv.weight", "decoder.13.conv.bias".

@Dwawayu
Copy link
Member

Dwawayu commented Mar 30, 2023

Im so sorry for forgetting to include the training flag --use_mixture_loss in the eval.sh😭, thank you for pointing out! You can add the flag --use_mixture_loss \ to eval.sh to fix it, additionally you may need to adjust the --batch_size to avoid OOM. You are welcome to reopen or create a new issue if you have any further questions or would like to discuss anything.

@Dwawayu Dwawayu closed this as completed Mar 30, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants