Skip to content

Commit 0d480d5

Browse files
committed
Add documentation about resume flag
1 parent 7fcc128 commit 0d480d5

1 file changed

Lines changed: 38 additions & 0 deletions

File tree

docs/pages/example_workflows/reinforcement_learning/step_2_policy_training.rst

Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,44 @@ To train the policy, run:
7373
- Record training videos periodically
7474
* - ``--video_interval 2000``
7575
- Interval for recording videos (iterations)
76+
* - ``--resume``
77+
- Resume training from a previous checkpoint (see :ref:`rl_resume_training` below)
78+
* - ``--load_run <str>``
79+
- Run folder to resume from (e.g. ``2026-01-29_12-30-00``); defaults to latest run in the experiment
80+
* - ``--checkpoint <str>``
81+
- Checkpoint file to load (e.g. ``model_2000.pt``); defaults to latest checkpoint in the run
82+
83+
84+
Resuming Training
85+
^^^^^^^^^^^^^^^^^
86+
.. _rl_resume_training:
87+
88+
To continue training from a saved checkpoint, use the ``--resume`` flag. Checkpoints are loaded from
89+
``logs/rsl_rl/generic_experiment/`` (or your ``--experiment_name``). Training continues from the
90+
loaded iteration; new checkpoints and logs are written to a new timestamped run directory.
91+
92+
**Resume from the latest run and latest checkpoint:**
93+
94+
.. code-block:: bash
95+
96+
python isaaclab_arena/scripts/reinforcement_learning/train.py \
97+
--resume \
98+
--headless \
99+
lift_object
100+
101+
**Resume from a specific run and checkpoint:**
102+
103+
.. code-block:: bash
104+
105+
python isaaclab_arena/scripts/reinforcement_learning/train.py \
106+
--resume \
107+
--load_run 2026-01-29_12-30-00 \
108+
--checkpoint model_2000.pt \
109+
--headless \
110+
lift_object
111+
112+
If ``--load_run`` or ``--checkpoint`` are omitted, the most recent run and checkpoint under the
113+
experiment directory are selected automatically.
76114

77115

78116
Training Configuration

0 commit comments

Comments
 (0)