Skip to content

Commit 3ff5694

Browse files
committed
Fix: #3833 Feedback about Multi GPU training with DDP
1 parent 5e98ccc commit 3ff5694

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

beginner_source/ddp_series_multigpu.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,8 @@ Running the distributed training job
202202
Here's what the code looks like:
203203

204204
.. code-block:: python
205+
206+
205207
def main(rank, world_size, total_epochs, save_every):
206208
ddp_setup(rank, world_size)
207209
dataset, model, optimizer = load_train_objs()

0 commit comments

Comments
 (0)