about GPU memory size

[universefall]

thanks for your work，my gpu is 2GB. but still shows out of memory. And I try to run it on CPU and change the GPU setting to FALSE. However, it cannot work.

[IlchaeJung]

Thank you for your interest in my work.
Unfortunately, the CPU version is not available currently.
I just make a CPU mode to implement in the future.
Actually, there is some trick to run this code by reducing the batch_size.
Please could you try this?

[universefall]

Yes, I tried to decrease the 'batch_test' from 256 to 2.However, still shows CUDA error: out of memory. My GPU is GTX1050.