RuntimeError: CUDA out of memory. Tried to allocate 25.00 MiB (GPU 0; 3.94 GiB total capacity; 3.14 GiB already allocated; 21.56 MiB free; 634.00 KiB cached)

[cherubicXN]

RuntimeError: CUDA out of memory. Tried to allocate 25.00 MiB (GPU 0; 3.94 GiB total capacity; 3.14 GiB already allocated; 21.56 MiB free; 634.00 KiB cached)

do you have this problem?how to deal with it?

Originally posted by @zhiyunpengpeng in https://github.com/cherubicXN/afm_cvpr2019/issues/4#issuecomment-492935657

[cherubicXN]

If you want to training the network, 4GB memory might be not enough.