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

Format import using isort --profile black --gitignore --lai 2 .

Browse Source
This commit is contained in:
litagin02
2024-03-11 12:37:51 +09:00
parent 164b5c4a85
commit dada0af2b6
29 changed files with 50 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_ms.py
View File

@@ -24,8 +24,7 @@ from data_utils import (
from losses import discriminator_loss, feature_loss, generator_loss, kl_loss
from mel_processing import mel_spectrogram_torch, spec_to_mel_torch
from style_bert_vits2.logging import logger
from style_bert_vits2.models import commons
from style_bert_vits2.models import utils
from style_bert_vits2.models import commons, utils
from style_bert_vits2.models.hyper_parameters import HyperParameters
from style_bert_vits2.models.models import (
DurationDiscriminator,
@@ -35,6 +34,7 @@ from style_bert_vits2.models.models import (
from style_bert_vits2.nlp.symbols import SYMBOLS
from style_bert_vits2.utils.stdout_wrapper import SAFE_STDOUT
torch.backends.cuda.matmul.allow_tf32 = True
torch.backends.cudnn.allow_tf32 = (
True # If encontered training problem,please try to disable TF32.