CaoWangrenbo 1369edaad7 feat: enable BatchNorm2d in CNNEncoder and add AMP support ...

- Uncomment BatchNorm2d in CNNEncoder (activated when cfg.use_bn=True)
- Add torch.amp.GradScaler + autocast for mixed precision training
- Add --amp/--no-amp CLI argument (default: enabled)

Generated by Mistral Vibe. deepseek-v4-flash
Co-Authored-By: Mistral Vibe <vibe@mistral.ai>

2026-06-08 00:41:34 +08:00

benchmark

initial commit

2026-05-29 18:49:01 +08:00

src

feat: enable BatchNorm2d in CNNEncoder and add AMP support

2026-06-08 00:41:34 +08:00

visualize

refactor: replace rotation vector with body up vector for tilt input

2026-06-04 21:02:08 +08:00

.gitignore

initial commit

2026-05-29 18:49:01 +08:00

AGENTS.md

feat: replace non-overlapping windows with sliding-window sequence sampling

2026-06-05 16:45:24 +08:00

batch_convert.sh

initial commit

2026-05-29 18:49:01 +08:00

DATASET_FORMAT.md

initial commit

2026-05-29 18:49:01 +08:00

download_davis_gt_rosbags.sh

initial commit

2026-05-29 18:49:01 +08:00

requirements.txt

initial commit

2026-05-29 18:49:01 +08:00

rosbag2wds.py

initial commit

2026-05-29 18:49:01 +08:00

start_ros_container.sh

fix: make start_ros_container.sh executable

2026-06-04 20:37:58 +08:00

Description

No description provided

134 KiB

Languages

Python 92.1%

Shell 7.9%