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')