Skip to content

Commit

Permalink
[Fix] Add the command in the README for the single-GPU case (#48)
Browse files Browse the repository at this point in the history
* Implement portable eval script for server

* lint format

* implement visualizer for debugging

* Update README

* docstring

* Refine some docstrings.

* refine docstring

* refine docstring

* update README for single gpu training/testing
  • Loading branch information
mxh1999 authored May 9, 2024
1 parent 0ee6e61 commit f564b57
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,10 @@ We provide configs for different tasks [here](configs/) and you can run the trai
For example, to train a multi-view 3D detection model with pytorch, just run:

```bash
# Single GPU training
python tools/train.py configs/detection/mv-det3d_8xb4_embodiedscan-3d-284class-9dof.py --work-dir=work_dirs/mv-3ddet

# Multiple GPU training
python tools/train.py configs/detection/mv-det3d_8xb4_embodiedscan-3d-284class-9dof.py --work-dir=work_dirs/mv-3ddet --launcher="pytorch"
```

Expand All @@ -190,6 +194,10 @@ NOTE: To run the multi-view 3D grounding experiments, please first download the
To inference and evaluate the model (e.g., the checkpoint `work_dirs/mv-3ddet/epoch_12.pth`), just run the test script:

```bash
# Single GPU testing
python tools/test.py configs/detection/mv-det3d_8xb4_embodiedscan-3d-284class-9dof.py work_dirs/mv-3ddet/epoch_12.pth

# Multiple GPU testing
python tools/test.py configs/detection/mv-det3d_8xb4_embodiedscan-3d-284class-9dof.py work_dirs/mv-3ddet/epoch_12.pth --launcher="pytorch"
```

Expand Down

0 comments on commit f564b57

Please sign in to comment.