PengCheng-NUDT

CUDA out of memory

2

Thanks for your great job! The se3-transformer is powerful, but seems to be memory exhaustive. I built a model with the following parameters, and got "CUDA out of memory error"...