游雁
2024-06-09 1186cd96a5a8fa3466c5e3e41f86f0280deb1410
funasr/bin/train_ds.py
@@ -180,7 +180,7 @@
            torch.cuda.empty_cache()
            time_escaped = (time.perf_counter() - time1) / 3600.0
            time_escaped = (time.perf_counter() - time_slice_i) / 3600.0
            logging.info(
                f"rank: {local_rank}, "
                f"time_escaped_epoch: {time_escaped:.3f} hours, "