diff --git a/configs/keypoint/README.md b/configs/keypoint/README.md index 286fe64dcc0752a4040fbf49a9fbb850a2754716..6076193cdcc39127ac2b47bcd1cb65b829fd42c9 100644 --- a/configs/keypoint/README.md +++ b/configs/keypoint/README.md @@ -50,24 +50,27 @@ MPII数据集 ​ **单卡训练:** ```shell +#COCO DataSet CUDA_VISIBLE_DEVICES=0 python3 tools/train.py -c configs/keypoint/higherhrnet/higherhrnet_hrnet_w32_512.yml #MPII DataSet -CUDA_VISIBLE_DEVICES=0 nohup python3 tools/train.py -c configs/keypoint/hrnet/hrnet_w32_256x256_mpii.yml +CUDA_VISIBLE_DEVICES=0 python3 tools/train.py -c configs/keypoint/hrnet/hrnet_w32_256x256_mpii.yml ``` ​ **多卡训练:** ```shell +#COCO DataSet CUDA_VISIBLE_DEVICES=0,1,2,3 python3 -m paddle.distributed.launch tools/train.py -c configs/keypoint/higherhrnet/higherhrnet_hrnet_w32_512.yml #MPII DataSet -CUDA_VISIBLE_DEVICES=0,1,2,3 nohup python3 -m paddle.distributed.launch tools/train.py -c configs/keypoint/hrnet/hrnet_w32_256x256_mpii.yml +CUDA_VISIBLE_DEVICES=0,1,2,3 python3 -m paddle.distributed.launch tools/train.py -c configs/keypoint/hrnet/hrnet_w32_256x256_mpii.yml ``` ​ **模型评估:** ```shell +#COCO DataSet CUDA_VISIBLE_DEVICES=0 python3 tools/eval.py -c configs/keypoint/higherhrnet/higherhrnet_hrnet_w32_512.yml #MPII DataSet