Can not resume training by giving ckpt

[tommy-qichang]

Hi there,

Thank you so much for your code and paper. Very impressive work.

I just want to reproduce your result, and so far so good.
But when I want to resume my ckpt from the previous training, it throws the following error.
load model: /research/cbim/vast/qc58/work/projects/semanticGAN/ckpt/run-Feb03_16-23-14/ckpt/060000.pt Traceback (most recent call last): File "train_seg_gan.py", line 679, in <module> g_optim.load_state_dict(ckpt['g_optim']) KeyError: 'g_optim'

I think there is no code save g_optim, d_img_optim and d_seg_optim when training, how can you load these setting from the checkpoint? Please let me know how you resume your training and if I missed anything?

Thanks again.

[ouyangshixiong]

I just comment three optim lines