search

samxuxiang / BrepGen

[SIGGRAPH 2024] Official PyTorch Implementation of "BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometry".
Other
227 stars 28 forks source link

硬件需求咨询 #9

Closed lower01 closed 4 months ago

lower01 commented 4 months ago

您好，请问这个项目大概需要多少显存的GPU支持呢？

samxuxiang commented 4 months ago

Hi @lower01 , for training, a minimum of four GPU with 24GB memory should work if you reduce the batch size for edge latent diffusion module.