SN-GAN
SN-GAN copied to clipboard
Is it unable to save self.u?
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!
Hi,
Thanks for find this issue. I just find self.u
is saved as None
.
Then I will figure it out .
@WellYoungIOE Sorry for disappear for a long time. I have fixed the self.u bug for inference.