VITA-Group/GNT

loaded state dict contains a parameter group that doesn't match the size of optimizer's group

leo-frank opened this issue · 1 comments

Descrpiton

I use the pretrained generalization model at https://drive.google.com/file/d/1AMN0diPeHvf2fw53IO5EE2Qp4os5SkoX/view?usp=share_link.
The command is
CUDA_VISIBLE_DEVICES=2 python3 eval.py --config configs/gnt_blender.txt \ --eval_dataset nerf_synthetic \ --eval_scenes mic --run_val \ --expname gnt_author_pretrained_cross_mic \ --ckpt_path /home/cbe/lwy/GNT/out/pretrained_cross_720000.pth

Bug

  File "/home/cbe/lwy/GNT/gnt/model.py", line 133, in load_from_ckpt
    self.load_model(fpath, load_opt, load_scheduler)
  File "/home/cbe/lwy/GNT/gnt/model.py", line 102, in load_model
    self.optimizer.load_state_dict(to_load["optimizer"])
  File "/home/cbe/miniconda3/envs/vision/lib/python3.8/site-packages/torch/optim/optimizer.py", line 146, in load_state_dict
    raise ValueError("loaded state dict contains a parameter group "
ValueError: loaded state dict contains a parameter group that doesn't match the size of optimizer's group

Questions

I wonder if you can run the pretrained generalization model ? or why I met such problem?

Related Issue

before renamed, #8 also met same problem when he/she runs render.py, but i am running eval.py.

Thanks for your great work !!!

sorry to bother you , i found the reason, it's because that the config file is not mismatched, should use --config configs/gnt_full.txt

Thanks for your great work again !!!