wzzheng / OccWorld

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

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

Closed liuziyang123 closed 9 months ago

liuziyang123 commented 9 months ago

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

SeaBird-Go commented 9 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 9 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 9 months ago

Many thanks.