feat(train): use RichProgressBar (#311)

voicepaw · Apr 13, 2023 · 17e937a · 17e937a
1 parent 200d47c
commit 17e937a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/so_vits_svc_fork/train.py b/src/so_vits_svc_fork/train.py
@@ -82,6 +82,7 @@ def train(
         if hparams.train.get("bf16_run", False)
         else 32,
         strategy=strategy,
+        callbacks=[pl.callbacks.RichProgressBar()],
     )
     model = VitsLightning(reset_optimizer=reset_optimizer, **hparams)
     trainer.fit(model, datamodule=datamodule)