deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-04 17:39:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

update lint

Browse Source
This commit is contained in:
lyuxiang.lx
2024-10-16 13:06:31 +08:00
parent 555efd0301
commit 29507bc77a
8 changed files with 21 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cosyvoice/utils/executor.py
View File

@@ -25,7 +25,7 @@ from cosyvoice.utils.train_utils import update_parameter_and_lr, log_per_step, l
class Executor:
def __init__(self, gan: bool=False):
def __init__(self, gan: bool = False):
self.gan = gan
self.step = 0
self.epoch = 0
@@ -81,7 +81,8 @@ class Executor:
dist.barrier()
self.cv(model, cv_data_loader, writer, info_dict, on_batch_end=True)
def train_one_epoc_gan(self, model, optimizer, scheduler, optimizer_d, scheduler_d, train_data_loader, cv_data_loader, writer, info_dict, group_join):
def train_one_epoc_gan(self, model, optimizer, scheduler, optimizer_d, scheduler_d, train_data_loader, cv_data_loader,
writer, info_dict, group_join):
''' Train one epoch
'''