About train_diff_mode
zxt-triumph opened this issue · 2 comments
File "/home/dell/ZXT/Code/DiffusionModels/DDM2/model/model.py", line 71, in init
self.load_network()
File "/home/dell/ZXT/Code/DiffusionModels/DDM2/model/model.py", line 224, in load_network
self.optG.load_state_dict(opt['optimizer'])
File "/home/dell/anaconda3/envs/pytorch/lib/python3.10/site-packages/torch/optim/optimizer.py", line 171, 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
I encountered the above problems when I run ./run_stage3.sh, how can I solve it?
Hi, this is probably due to an incorrect setting of the path for either stage1 trained model or stage2 matched file. Can you please double-check the paths here are correctly entered?
Thank you for your reply,! This problem has been solved under your prompt.