Lee1413 commited on
Commit
dcaf240
·
verified ·
1 Parent(s): 08ed2ee

Correct training provenance, add code link, config with loss-weight keys

Browse files
Files changed (2) hide show
  1. README.md +4 -3
  2. mapqr_mapgr_nusc_r50_24ep_config.py +4 -1
README.md CHANGED
@@ -14,7 +14,7 @@ datasets:
14
 
15
  This repository hosts the released weights for **Learning Global Representation from Queries for Vectorized HD Map Construction** (ICML 2026, [arXiv:2510.06969](https://arxiv.org/abs/2510.06969), [project page](https://xr-lee.github.io/mapgr-demo/)).
16
 
17
- Training and evaluation code, installation and data preparation are in the GitHub repository (`mapqr/` sub-project).
18
 
19
  | File | Description |
20
  |---|---|
@@ -30,11 +30,12 @@ Training and evaluation code, installation and data preparation are in the GitHu
30
 
31
  mAP<sub>2</sub> averages Chamfer AP over the {0.5, 1.0, 1.5} m thresholds; mAP<sub>1</sub> averages over {0.2, 0.5, 1.0} m.
32
 
33
- This checkpoint was **re-trained with the released code**. It is not the run reported in the paper.
34
 
 
35
  - Training: 8 GPUs × 4 samples, fp16.
36
  - The numbers above come from evaluating this exact file with the released `tools/test.py`.
37
- - Repeated runs of this setting vary by roughly ±0.4 mAP<sub>2</sub>.
38
 
39
  ```shell
40
  # inside mapqr/ of the code repository
 
14
 
15
  This repository hosts the released weights for **Learning Global Representation from Queries for Vectorized HD Map Construction** (ICML 2026, [arXiv:2510.06969](https://arxiv.org/abs/2510.06969), [project page](https://xr-lee.github.io/mapgr-demo/)).
16
 
17
+ Training and evaluation code, installation and data preparation are in the GitHub repository [XR-Lee/MapGR](https://github.com/XR-Lee/MapGR) (`mapqr/` sub-project).
18
 
19
  | File | Description |
20
  |---|---|
 
30
 
31
  mAP<sub>2</sub> averages Chamfer AP over the {0.5, 1.0, 1.5} m thresholds; mAP<sub>1</sub> averages over {0.2, 0.5, 1.0} m.
32
 
33
+ This checkpoint is not the run reported in the paper.
34
 
35
+ - It was trained with the research code the GitHub repository was cleaned up from, using the same settings as the released config.
36
  - Training: 8 GPUs × 4 samples, fp16.
37
  - The numbers above come from evaluating this exact file with the released `tools/test.py`.
38
+ - Other 24-epoch runs of closely related configurations (different seeds, loss weights and query counts) reached 67.2–68.0 mAP<sub>2</sub>.
39
 
40
  ```shell
41
  # inside mapqr/ of the code repository
mapqr_mapgr_nusc_r50_24ep_config.py CHANGED
@@ -452,7 +452,10 @@ model = dict(
452
  loss_pts=dict(type='PtsL1Loss', loss_weight=5.0),
453
  loss_dir=dict(type='PtsDirCosLoss', loss_weight=0.005),
454
  loss_seg=dict(type='SimpleLoss', pos_weight=4.0, loss_weight=1.0),
455
- loss_pv_seg=dict(type='SimpleLoss', pos_weight=1.0, loss_weight=2.0)),
 
 
 
456
  train_cfg=dict(
457
  pts=dict(
458
  grid_size=[512, 512, 1],
 
452
  loss_pts=dict(type='PtsL1Loss', loss_weight=5.0),
453
  loss_dir=dict(type='PtsDirCosLoss', loss_weight=0.005),
454
  loss_seg=dict(type='SimpleLoss', pos_weight=4.0, loss_weight=1.0),
455
+ loss_pv_seg=dict(type='SimpleLoss', pos_weight=1.0, loss_weight=2.0),
456
+ aux_loss_weights=[0.05, 0.05, 1.0, 1.0, 1.0],
457
+ aux_loss_weights_one2many=[1.0, 1.0, 1.0, 1.0, 1.0],
458
+ global_map_loss_weights=[1.0, 1.0, 0.0, 0.0, 0.0, 0.0]),
459
  train_cfg=dict(
460
  pts=dict(
461
  grid_size=[512, 512, 1],