def pre_train(self): embedding_optim_cfg = OptimConfig('Adam', lr=1e-2) self.embedding_optim = embedding_optim_cfg.get( self.gen.embedding.parameters()) embedding_loss_cfg = LossConfig('MSE') self.embedding_loss = embedding_loss_cfg.get() self.gen_optim = self.goptim_cfg.get( filter(lambda x: type(x) != SampleMatrix, self.gen.parameters()))
def pre_train(self): embedding_optim_cfg = OptimConfig('Adam', lr=1e-3) self.embedding_optim = embedding_optim_cfg.get( self.gen.embedding.parameters()) embedding_loss_cfg = LossConfig('MSE') self.embedding_loss = embedding_loss_cfg.get() self.gen_optim = goptim_cfg.get( filter(lambda x: type(x) != SampleMatrix, self.gen.parameters())) logger.create_scalar('js') logger.create_scalar('coverage') logger.create_scalar('coverage_HQ') logger.create_scalar('confidence') logger.create_scalar('ratio')