OpenDriveLab / ViDAR

[CVPR 2024 Highlight] Visual Point Cloud Forecasting
https://arxiv.org/abs/2312.17655
Apache License 2.0
278 stars 18 forks source link

修改哪些参数可以保证用较少的GPU达到训练模型的地步 #39

Closed naonao-wu closed 3 months ago

naonao-wu commented 3 months ago

你好,因为我的可用GPU较少,请问除了修改supervise_all_future=False,vidar_head_pred_history_frame_num=0,vidar_head_pred_future_frame_num=0,andvidar_head_per_frame_loss_weight=(1.0,)这几个参数外,还可以修改哪些参数?从而保证用较少的GPU达到训练模型的地步。 我试了一些参数,但是感觉还是没有找到能够起到本质作用的参数。 期待你的回复。

tomztyang commented 3 months ago

感谢您的关注。可以参考Issue 17的一些解决方案。

但是我们不保证3090可以复现文章中的结果,我们之前都是用a100做的实验。

祝好, 杨泽同