speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add

Browse Source
This commit is contained in:
游雁 2024-08-09 11:12:52 +08:00
parent 27622202a6
commit cf85b11448
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/train_utils/trainer_ds.py
View File

@ -145,7 +145,7 @@ class Trainer:
try:
from tensorboardX import SummaryWriter
self.writer = SummaryWriter(tensorboard_dir) # if trainer.rank == 0 else None
self.writer = SummaryWriter(tensorboard_dir) if self.rank == 0 else None
except:
self.writer = None
@ -190,7 +190,7 @@ class Trainer:
step_in_epoch = None if step is None else step_in_epoch
if self.use_deepspeed:
logging.info(f"Save checkpoint: {epoch}, rank: {self.local_rank}\n")
logging.info(f"Save checkpoint: {epoch}, rank: {self.rank}\n")
# self.step_or_epoch += 1
state = {
"epoch": epoch,