mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
removed logging
This commit is contained in:
1 parent
229d168c20
commit
10e031a843
1 file changed
+3
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user