search

vzyrianov / lidargen

Official implementation of "Learning to Generate Realistic LiDAR Point Clouds" (ECCV 2022)
https://www.zyrianov.org/lidargen/
MIT License
135 stars 14 forks source link

Train on RTX A4000 #9

Open Beastmaster opened 1 year ago

Beastmaster commented 1 year ago

Thanks for sharing your code! While I have some questions. How did you manage to train the model on A4000(I suppose 16G)? The bach size on "/workspace/third/lidargen/kitti.yml" is 24. I can only set to batch=4 on 3090GPU (24G). Did I miss something?

BTW, I trained kitti with batch=4, but got unstatisfying results.

HaiyunW commented 8 months ago

@Beastmaster Hello, I trained kitti-360 with bachsize=4 and got unstatisfying results, too. Do you reslove this problem? I want to retry with bigger bachsize, but I‘m not sure if it will works.

Beastmaster commented 8 months ago

No. I didn't fix it. I compared pretrained model with my model, I got similar results(visually). I forgot exact numbers of evaluation metric.