游雁
2024-06-09 b75d1e89bb2f513a79bb07e9100ba1cd2bbcf40c
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, "