search

StanfordMIMI / DDM2

[ICLR2023] Official repository of DDM2: Self-Supervised Diffusion MRI Denoising with Generative Diffusion Models
125 stars 20 forks source link

About train_diff_mode #2

Closed zxt-triumph closed 1 year ago

zxt-triumph commented 1 year ago

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?

tiangexiang commented 1 year ago

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?

image
zxt-triumph commented 1 year ago

Thank you for your reply,! This problem has been solved under your prompt.