how many epochs to run to reproduce your results on waymo?

zhejz / HPTR

Real-Time Motion Prediction via Heterogeneous Polyline Transformer with Relative Pose Encoding. NeurIPS 2023.

https://zhejz.github.io/hptr

Other

120 stars 9 forks source link

how many epochs to run to reproduce your results on waymo? #4

Closed SixingHu closed 9 months ago

SixingHu commented 9 months ago

Thank u for open sourcing your great work.

May i know how many epochs are needed to reproduce your results on waymo? In your HPTR/configs/trainer/womd.yaml file, what hyper parameter should be set? max_epochs: null min_epochs: null

Thank you in advance for you kind help!

zhejz commented 9 months ago

Hi, you can find the training details in Sec 4.1 in our paper. Our final models are trained for 120 epochs for WOMD and 150 epochs for AV2.

SixingHu commented 9 months ago

Thank you for your kind help!