Residual extraction network is never updated in joint training stage

[biubiubiiu]

Code snippet from train_rain_class.py:

netG = net1.vgg19ca()
residue_net = net2.Dense_rain_residual()
# ...
optimizerG = optim.Adam(netG.parameters(), ...)
# ...
optmizerG.step()

It seems that train_rain_class.py corresponds to the joint optimization stage in your paper, but only the density classifier network is updated here. Is it the expected behavior?

BTW, there is an undefined variable optimizer_D in line 156. I'm not sure what it's for.