From 5669c398fd4b9e8a0373105508677d423f86c5f4 Mon Sep 17 00:00:00 2001 From: Chao Fan <92680886+ChaoFan996@users.noreply.github.com> Date: Mon, 31 Oct 2022 16:51:53 +0800 Subject: [PATCH] Update 0.get_started.md --- docs/0.get_started.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/0.get_started.md b/docs/0.get_started.md index 8e03383..a44d0f6 100644 --- a/docs/0.get_started.md +++ b/docs/0.get_started.md @@ -37,7 +37,7 @@ See [prepare dataset](2.prepare_dataset.md). ## Train Train a model by ``` -CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/baseline/baseline.yaml --phase train +CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./configs/baseline/baseline.yaml --phase train ``` - `python -m torch.distributed.launch` [DDP](https://pytorch.org/tutorials/intermediate/ddp_tutorial.html) launch instruction. - `--nproc_per_node` The number of gpus to use, and it must equal the length of `CUDA_VISIBLE_DEVICES`. @@ -51,7 +51,7 @@ You can run commands in [train.sh](train.sh) for training different models. ## Test Evaluate the trained model by ``` -CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/baseline/baseline.yaml --phase test +CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./configs/baseline/baseline.yaml --phase test ``` - `--phase` Specified as `test`. - `--iter` Specify a iteration checkpoint.