update

游雁

2024-05-17 b85e140c3e4a7a7ccba59abfc67b63aac7a28dd9

update

 funasr/train_utils/trainer_ds.py

@@ -398,7 +398,7 @@
                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}"