memory usage ，too

[pppkkk611]

Hi Thank you for your nice work~
I want to check with you the memory usage of the program. When I run on my 11G gpu, I can only set output point cloud to 1024, with a batch size=1. Bigger than that will raise OOM issue.
The cd loss function takes a lot memory, for batch size=1 is 4x2000x5000x2, So I cannot train any model to generate 8000 points, not to mention 16000 points.
How do you deal with this memory issue?
Thank you very much.