SN-GAN icon indicating copy to clipboard operation
SN-GAN copied to clipboard

Is it unable to save self.u?

Open WellYoungIOE opened this issue 7 years ago • 2 comments

The self.u in SNCon2d seems unable to save with torch.save(SND.state_dict(), '%s/netD_epoch_%d.pth' % ('log', epoch)) Then how can I use the learned u to do inference? Thanks!

WellYoungIOE avatar Dec 06 '17 13:12 WellYoungIOE

Hi, Thanks for find this issue. I just find self.u is saved as None. Then I will figure it out .

godisboy avatar Dec 07 '17 00:12 godisboy

@WellYoungIOE Sorry for disappear for a long time. I have fixed the self.u bug for inference.

godisboy avatar Mar 12 '18 06:03 godisboy