From 89632bf748f90565e2c75b6aeb6c6460309713d5 Mon Sep 17 00:00:00 2001 From: jinzr Date: Fri, 8 Mar 2024 02:56:53 +0800 Subject: [PATCH] Update train.py --- egs/vctk/TTS/vits/train.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/egs/vctk/TTS/vits/train.py b/egs/vctk/TTS/vits/train.py index 9b2f2b246..d65dfd5c3 100755 --- a/egs/vctk/TTS/vits/train.py +++ b/egs/vctk/TTS/vits/train.py @@ -514,9 +514,10 @@ def train_one_epoch( logging.warning(f"Grad scale is small: {cur_grad_scale}") if cur_grad_scale < 1.0e-05: save_bad_model() - raise RuntimeError( - f"grad_scale is too small, exiting: {cur_grad_scale}" - ) + logging.warning(f"Grad scale is small: {cur_grad_scale}") + # raise RuntimeError( + # f"grad_scale is too small, exiting: {cur_grad_scale}" + # ) if params.batch_idx_train % params.log_interval == 0: cur_lr_g = max(scheduler_g.get_last_lr())