search

Rgtemze / HyperDiffusion

Official code repository of "HyperDiffusion: Generating Implicit Neural Fields with Weight-Space Diffusion" @ ICCV 2023
Other
170 stars 16 forks source link

run question #9

Open lvhyang opened 1 month ago

lvhyang commented 1 month ago

Great job, but may I ask how much memory is needed to run the training code? I encountered a low memory error on two 4090 cards.

微信图片_20240801095316
ehedlin commented 1 month ago

I find that it requires just below 32 gigs of VRAM to run. If you want to reduce the memory requirements you can make the hypernetwork smaller (it currently has ~1.3 billion params)