From d94b48e0f421afa8c09f5cb07aad43a1e3ef3ba1 Mon Sep 17 00:00:00 2001 From: Fenghe Tang <54798078+FengheTan9@users.noreply.github.com> Date: Thu, 14 Mar 2024 14:57:02 +0800 Subject: [PATCH] Update readme.md --- readme.md | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/readme.md b/readme.md index 9efef8d..7f05262 100644 --- a/readme.md +++ b/readme.md @@ -122,13 +122,17 @@ python -m datasets.cache_dataset3d ## Start Training -Run training on multi-GPU +Run training on multi-gpu ``` CUDA_VISIBLE_DEVICES=0,1,2,3 python -m core.ddp --tag debug ``` +Run training on single-GPU +``` +CUDA_VISIBLE_DEVICES=0 python -m core.ddp --tag debug +``` ## Sliding Inference and Test