OOM Errors when running EnamineHTS_single_batch.ini on a machine with RTX3090

[WangyangBit]

Thanks for great work!

I'm trying to reproduce experimental results of EnamineHTS but was unable to deal with the CUDA OOM error in inference step.

I have tried methods in #47 however the error is still here.

Here is my screenshot of nvidia-smi:

and here are tracebacks of the error:

I have no idea on how to deal with this.