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

log step

Browse Source
This commit is contained in:
游雁 2024-05-01 01:05:59 +08:00
parent f5958e96da
commit f690741b9a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
funasr/train_utils/trainer.py
View File

@ -465,7 +465,8 @@ class Trainer:
batch_num_epoch = len(dataloader_train) batch_num_epoch = len(dataloader_train)
self.log( self.log(
epoch, epoch,
batch_idx + kwargs.get("start_step", 0), batch_idx,
log_step=batch_idx + kwargs.get("start_step", 0),
step_in_epoch=self.step_in_epoch, step_in_epoch=self.step_in_epoch,
batch_num_epoch=batch_num_epoch, batch_num_epoch=batch_num_epoch,
lr=lr, lr=lr,
@ -634,11 +635,12 @@ class Trainer:
tag="train", tag="train",
data_split_i=0, data_split_i=0,
data_split_num=1, data_split_num=1,
log_step=None,
**kwargs, **kwargs,
): ):
if (batch_idx + 1) % self.log_interval == 0: if (batch_idx + 1) % self.log_interval == 0:
batch_idx = log_step if log_step is not None else batch_idx
gpu_info = ( gpu_info = (
"GPU, memory: usage: {:.3f} GB, " "GPU, memory: usage: {:.3f} GB, "
"peak: {:.3f} GB, " "peak: {:.3f} GB, "