kirigaya/Slide-SAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update readme.md

Browse Source
This commit is contained in:
Fenghe Tang 2024-03-14 14:57:02 +08:00 committed by GitHub
parent b975f29212
commit d94b48e0f4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
readme.md
View File

@ -122,13 +122,17 @@ python -m datasets.cache_dataset3d
## Start Training ## 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 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 ## Sliding Inference and Test