tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update train_ms.py

Browse Source
This commit is contained in:
Stardust·减
2023-11-25 17:30:24 +08:00
committed by GitHub
parent 65cf40fa49
commit f65b332f93
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train_ms.py
View File

@@ -1,5 +1,4 @@
# flake8: noqa: E402 # flake8: noqa: E402
import platform import platform
import os import os
import torch import torch
@@ -582,7 +581,7 @@ def train_and_evaluate(
epoch, epoch,
os.path.join(hps.model_dir, "DUR_{}.pth".format(global_step)), os.path.join(hps.model_dir, "DUR_{}.pth".format(global_step)),
) )
keep_ckpts = getattr(hps.train, "keep_ckpts", 10) keep_ckpts = config.train_ms_config.keep_ckpts
if keep_ckpts > 0: if keep_ckpts > 0:
utils.clean_checkpoints( utils.clean_checkpoints(
path_to_models=hps.model_dir, path_to_models=hps.model_dir,