removed logging

2026-09-11 12:31:23 +08:00 · 2019-07-20 09:17:20 -04:00
parent 229d168c20
commit 10e031a843
1 changed files with 3 additions and 0 deletions
@@ -380,6 +380,9 @@ class Trainer(TrainerIO):
            # otherwise we launch the required number of processes
            try:
                nb_slurm_tasks = int(os.environ['SLURM_NTASKS'])
+                print('x'*100)
+                print(f'NB_TASKS: {nb_slurm_tasks}')
+                print('x'*100)
                nb_requested_gpus = len(self.data_parallel_device_ids)
                is_slurm_managing_tasks = nb_slurm_tasks == nb_requested_gpus
            except Exception as e: