RuntimeError: CUDA out of memory.
biotech70 opened this issue · 1 comments
biotech70 commented
RuiWang1998 commented
Hi,
Unfortunately it might be not possible to run the model on a 2GB-ram card. You could try to set --subbatch_size
to 1 and see if it runs. If it does not run and output the same error stack, then we suggest you get access to a larger card (through colab, for example).
Best