docs: add uv workflow and ScoNet eval reproduction notes
This commit is contained in:
@@ -76,6 +76,20 @@ See [here](https://github.com/jdyjjj/All-in-One-Gait) for details.
|
||||
- **Nice log**: We use [`tensorboard`](https://pytorch.org/docs/stable/tensorboard.html) and `logging` to log everything, which looks pretty.
|
||||
|
||||
## Getting Started
|
||||
### Quick Start (uv)
|
||||
```bash
|
||||
# Install dependencies
|
||||
uv sync --extra torch
|
||||
|
||||
# Train
|
||||
CUDA_VISIBLE_DEVICES=0,1 uv run python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./configs/baseline/baseline.yaml --phase train
|
||||
|
||||
# Test
|
||||
CUDA_VISIBLE_DEVICES=0,1 uv run python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./configs/baseline/baseline.yaml --phase test
|
||||
```
|
||||
|
||||
> **Note:** The `--nproc_per_node` argument must exactly match the number of GPUs specified in `CUDA_VISIBLE_DEVICES`. For single-GPU evaluation, use `CUDA_VISIBLE_DEVICES=0` and `--nproc_per_node=1` with the DDP launcher.
|
||||
|
||||
|
||||
|
||||
Please see [0.get_started.md](docs/0.get_started.md). We also provide the following tutorials for your reference:
|
||||
|
||||
Reference in New Issue
Block a user