SIREN memory requirements

[pennywise86hw]

What are the minimal memory requirements for train_wave_equation.py? I am trying to run it on my gpu with 16GB memory, but consistently getting cuda OOM..

[MahirGulzar]

I am also getting the same error tried both 4GB and 16GB GPUs. Any solution?

[vishwa91]

Comment out the lines where you the spatial gradients are being computed for visualization. That reduced my memory usage to less than 4GB.

[ardywibowo]

Could you elaborate more? I couldn't find any lines that were not necessary for the loss computation.

[ardywibowo]

I've found it. Simply comment out summary_fn(...) in training.py. Something in write_helmholtz_summary(...) is eating a lot of memory.