From f49c4ee4b561ed2cee40940d4848dac635f3a24b Mon Sep 17 00:00:00 2001 From: Abdul Fatir Date: Fri, 7 Jun 2024 23:36:45 +0200 Subject: [PATCH] Remove print statements from `train.py` (#101) *Description of changes:* Removes print statements that got left inside from a debugging session. By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. --- scripts/training/train.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/scripts/training/train.py b/scripts/training/train.py index ada4a99..d14aa75 100644 --- a/scripts/training/train.py +++ b/scripts/training/train.py @@ -72,24 +72,17 @@ def get_training_job_info() -> Dict: # CUDA info job_info["cuda_available"] = torch.cuda.is_available() - print(job_info) if torch.cuda.is_available(): job_info["device_count"] = torch.cuda.device_count() - print(job_info) job_info["device_names"] = { idx: torch.cuda.get_device_name(idx) for idx in range(torch.cuda.device_count()) } - print(job_info) - - print(torch.cuda.mem_get_info(device=0)) - print(torch.cuda.mem_get_info(device=1)) job_info["mem_info"] = { idx: torch.cuda.mem_get_info(device=idx) for idx in range(torch.cuda.device_count()) } - print(job_info) # DDP info job_info["torchelastic_launched"] = dist.is_torchelastic_launched()