added analysis notebook

2026-09-24 13:50:25 +08:00 · 2019-07-21 12:18:46 -04:00
parent 7ac344e43a
commit 7e053fc731
1 changed files with 1 additions and 1 deletions
@@ -393,9 +393,9 @@ class Trainer(TrainerIO):

            # whenever we have the correct number of tasks, we let slurm manage processes
            # otherwise we launch the required number of processes
+            nb_requested_gpus = len(self.data_parallel_device_ids) * self.nb_gpu_nodes
            try:
                nb_slurm_tasks = int(os.environ['SLURM_NTASKS'])
-                nb_requested_gpus = len(self.data_parallel_device_ids) * self.nb_gpu_nodes
                is_slurm_managing_tasks = nb_slurm_tasks == nb_requested_gpus
            except Exception as e:
                # likely not on slurm, so set the slurm managed flag to false