Skip to content

Commit 47ed228

Browse files
authored
Update README.md
1 parent d634cc8 commit 47ed228

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

README.md

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,7 @@ If you found this paper or codebase useful, please cite our paper:
3131
# Run VectorMapNet
3232

3333
## Note
34-
We are still working on refining this repo. Currently, we only release the inference code of VectorMapNet.
3534

36-
The training pipeline will be released in next week.
3735

3836
## 0. Environment
3937

@@ -66,6 +64,20 @@ python tools/data_converter/nuscenes_converter.py --data-root your/dataset/nuSce
6664
|--------------|-------------|--------|------------|
6765
| VectorMapNet | Camera only | [config](configs/vectormapnet.py) | [model link](https://drive.google.com/file/d/1ccrlZ2HrFfpBB27kC9DkwCYWlTUpgmin/view?usp=sharing) |
6866

67+
68+
### Train VectorMapNet
69+
70+
In single GPU
71+
```
72+
python tools/train.py configs/vectormapnet.py
73+
```
74+
75+
For multi GPUs
76+
```
77+
bash tools/dist_train.sh configs/vectormapnet.py $num_gpu
78+
```
79+
80+
6981
### Do Evaluation
7082

7183
In single GPU

0 commit comments

Comments
 (0)