Skip to content

Commit 0e432b5

Browse files
committed
config: add aug in deltaflow training command.
1 parent 813e282 commit 0e432b5

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ Note (2025/09/18): We got accepted by NeurIPS 2025 and it's **spotlighted**!
1818
- [x] 2025/09/25: DeltaFlow Model file, config file and loss function. Update quick training example.
1919
- [x] 2025/09/29: Pre-trained weights for Argoverse 2, Waymo, nuScenes. _Contact me if any issue (e.g., ask for delete ckpt as privacy concern etc)._ These models are provided for research and reproducibility purposes only.
2020
- [x] Public review comments for readers to refer to future improvement/directions etc. Refer discussion [here](https://github.com/Kin-Zhang/DeltaFlow/discussions/2).
21-
- [ ] Merged into [OpenSceneFlow](https://github.com/KTH-RPL/OpenSceneFlow), check pull request here: https://github.com/KTH-RPL/OpenSceneFlow/pull/21
21+
- [x] Merged into [OpenSceneFlow](https://github.com/KTH-RPL/OpenSceneFlow), check pull request here: https://github.com/KTH-RPL/OpenSceneFlow/pull/21
2222

2323
## Quick Run
2424

@@ -37,7 +37,7 @@ unzip demo-data-v2.zip -d /home/kin/data/av2/h5py # to your data path
3737

3838
3. Run the training with the following command (modify the data path accordingly):
3939
```bash
40-
python train.py model=deltaflow loss_fn=deltaflowLoss batch_size=4 num_frames=5 voxel_size="[0.15,0.15,0.15]" point_cloud_range="[-38.4,-38.4,-3,38.4,38.4,3]" optimizer.lr=2e-4 train_data=${demo_train_data_path} val_data=${demo_val_data_path}
40+
python train.py model=deltaflow loss_fn=deltaflowLoss batch_size=4 num_frames=5 train_aug=True voxel_size="[0.15,0.15,0.15]" point_cloud_range="[-38.4,-38.4,-3,38.4,38.4,3]" optimizer.lr=2e-4 train_data=${demo_train_data_path} val_data=${demo_val_data_path}
4141
```
4242
### Evaluation
4343

conf/config.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ wandb_mode: disabled # [offline, disabled, online]
99
wandb_project_name: seflow
1010

1111
train_data: /home/kin/data/av2/h5py/demo/train
12+
train_aug: False
1213
val_data: /home/kin/data/av2/h5py/demo/val
1314

1415
output: ${model.name}-${slurm_id}

0 commit comments

Comments
 (0)