Closed ghost closed 3 years ago
In solver_encoder.py I added: torch.save({'model': self.G.state_dict(), 'optimizer': self.g_optimizer.state_dict()}, 'autovc_bib.ckpt') But file size is 242MB. Is you use different model? why size different?
torch.save({'model': self.G.state_dict(), 'optimizer': self.g_optimizer.state_dict()}, 'autovc_bib.ckpt')
In solver_encoder.py I added:
torch.save({'model': self.G.state_dict(), 'optimizer': self.g_optimizer.state_dict()}, 'autovc_bib.ckpt')
But file size is 242MB. Is you use different model? why size different?