wzzheng / OccWorld

[ECCV 2024] 3D World Model for Autonomous Driving
https://wzzheng.net/OccWorld/
Apache License 2.0
380 stars 25 forks source link

您好,我想问一下关于训练VQ-VAE的问题 #5

Closed liuziyang123 closed 11 months ago

liuziyang123 commented 11 months ago

您好,感谢您们出色的工作 我想问一下第一阶段VQ-VAE的训练细节,或者是否有相关的config文件?

SeaBird-Go commented 11 months ago

@chen-wl20 @wzzheng ,also want to ask this question, could you provide the config to train the VQ-VAE?

BTW, in your offered occworld.py config, the num_frames=15, while the end_frame=11, which could cause assert error in this line, could you help me to solve this?

Thanks in advance.

chen-wl20 commented 11 months ago

Thank you for your interest! We have update the code and the config files just now. Thank you for your continued attention. Feel free to star our project and follow subsequent updates!

liuziyang123 commented 11 months ago

Many thanks.