in main.py [0:0]
def run(self): if self.cfg.loss == 'kl': self.train_kl() elif self.cfg.loss == 'likelihood': self.train_likelihood() else: assert False