def run()

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