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

update

Browse Source
This commit is contained in:
游雁 2024-05-17 17:28:56 +08:00
parent 1ca314955f
commit b85e140c3e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/train_utils/trainer_ds.py
View File

@ -398,7 +398,7 @@ class Trainer:
loss_dict["speed_stats"]["backward_time"] = f"{time4 - time3:0.3f}"
self.update_step(model, optim, scheduler, scaler, loss_dict=loss_dict)
total_time = f"{(time.perf_counter() - time5) / accum_grad:0.3f}"
total_time = f"{(time.perf_counter() - time5):0.3f}"
time5 = time.perf_counter()
loss_dict["speed_stats"]["optim_time"] = f"{time5 - time4:0.3f}"