Minimun VRAM required

[Isabel-jin]

Thanks for your helpful work!
I tried your 2d-gaussian-monitor on RTX3090 GPU which has 24GB VRAM. However when I viewed the trained module , I encouted an error as follow. And when I viewed the training process, the program collapsed after some iterations. I think maybe the limitation of VRAM caused the problem above, and I wonder how much VRAM is required at least for this monitor due to your tests. Or does anything else caused these problems?