issues
search
Tsinghua-MARS-Lab
/
StateTransformer
209
stars
18
forks
source link
Rebuild preprocess module and abstract gptmodel class
#104
Closed
Shiduo-zh
closed
1 year ago
Shiduo-zh
commented
1 year ago
rebuild preprocess module into nuplan_rasterize, pdm_vectorize and waymo_preprocess.
rebuild TrajectoryGPT model class, and abstract the base model class.
add pdm_vectorize preprocess file, but there is still something to do to debug it.
add NuplanVectorEncoder class definition, details to be done.
rebuild rasterize encoder as multi-choice resnet