train_diffusion_model
ZhengJianwei2 opened this issue · 1 comments
i meet a same problem as a closed issues as follows:
Traceback (most recent call last):
File "train_diff_model.py", line 54, in
diffusion = Model.create_model(opt)
File "/home/star/jiangjiawei/DDM2-main/model/init.py", line 7, in create_model
m = M(opt)
File "/home/star/jiangjiawei/DDM2-main/model/model.py", line 71, in init
self.load_network()
File "/home/star/jiangjiawei/DDM2-main/model/model.py", line 223, in load_network
self.optG.load_state_dict(opt['optimizer'])
File "/home/star/anaconda3/envs/py37/lib/python3.7/site-packages/torch/optim/optimizer.py", line 201, 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
As reminded, i checked the config file path of stage 1 and stage 2， and i think the path setting is correct, are there any other problems may cause this issue? and the details of training logs and the .txt file are also attached.
I found the problem, in stage3, we need to set the resume _ stage in the path to null
