def main(cfg): trainer = pl.Trainer(**cfg.trainer) exp_manager(trainer, cfg.get("exp_manager", None)) model = HifiGanModel(cfg=cfg.model, trainer=trainer) epoch_time_logger = LogEpochTimeCallback() trainer.callbacks.extend([epoch_time_logger]) trainer.fit(model)
def hifigan_model(): test_root = os.path.dirname(os.path.abspath(__file__)) model = HifiGanModel.from_pretrained(model_name="tts_hifigan") return model
def main(cfg): trainer = pl.Trainer(**cfg.trainer) exp_manager(trainer, cfg.get("exp_manager", None)) model = HifiGanModel(cfg=cfg.model, trainer=trainer) model.maybe_init_from_pretrained_checkpoint(cfg=cfg) trainer.fit(model)
def hifigan_model(): model = HifiGanModel.from_pretrained(model_name="tts_hifigan") return model
def main(cfg): trainer = pl.Trainer(**cfg.trainer) exp_manager(trainer, cfg.get("exp_manager", None)) model = HifiGanModel(cfg=cfg.model, trainer=trainer) trainer.fit(model)