UNITER
UNITER copied to clipboard
Is it possible to extract features using 11GB GPU?
trafficstars
Now, the process reports F0625 08:25:59.166216 119 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory on my 11GB 2080ti. But that's all we have.
When I load the model into GPU, it costs only 1.8G, but the feature extraction requires way larger than that.