mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
added analysis notebook
This commit is contained in:
1 parent
7e053fc731
commit
0479784e7b
1 file changed
+1
@@ -394,6 +394,7 @@ 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
|
||||
nb_slurm_tasks = 0
|
||||
try:
|
||||
nb_slurm_tasks = int(os.environ['SLURM_NTASKS'])
|
||||
is_slurm_managing_tasks = nb_slurm_tasks == nb_requested_gpus
|
||||
|
||||
Reference in new issue
Block a user