Skip to content

Commit fdb03b2

Browse files
committed
Move documentation of new flag
Signed-off-by: Max Ehrlich <[email protected]>
1 parent 0dced67 commit fdb03b2

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

docs/source-pytorch/clouds/cluster_advanced.rst

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -88,18 +88,6 @@ To train a model using multiple nodes, do the following:
8888
8989
#SBATCH --signal=SIGUSR1@90
9090
91-
You can change this signal if your environment requires the use of a different one, for example
92-
93-
.. code-block:: bash
94-
95-
#SBATCH --signal=SIGHUP@90
96-
97-
Then, when you make your trainer, pass the `requeue_signal` option to the :class:`~pytorch_lightning.plugins.environments.slurm_environment.SLURMEnvironment` plugin:
98-
99-
.. code-block:: python
100-
101-
trainer = Trainer(plugins=[SLURMEnvironment(requeue_signal=signal.SIGHUP)])
102-
10391
6. Submit the SLURM job
10492

10593
.. code-block:: bash
@@ -125,6 +113,18 @@ To get this behavior make sure to add the correct signal to your SLURM script
125113
# 90 seconds before training ends
126114
SBATCH --signal=SIGUSR1@90
127115
116+
You can change this signal if your environment requires the use of a different one, for example
117+
118+
.. code-block:: bash
119+
120+
#SBATCH --signal=SIGHUP@90
121+
122+
Then, when you make your trainer, pass the `requeue_signal` option to the :class:`~pytorch_lightning.plugins.environments.slurm_environment.SLURMEnvironment` plugin:
123+
124+
.. code-block:: python
125+
126+
trainer = Trainer(plugins=[SLURMEnvironment(requeue_signal=signal.SIGHUP)])
127+
128128
If auto-resubmit is not desired, it can be turned off in the :class:`~pytorch_lightning.plugins.environments.slurm_environment.SLURMEnvironment` plugin:
129129

130130
.. code-block:: python

0 commit comments

Comments
 (0)