siewerthug

GPU memory allocation

5

Hi Whenever I try to run the following code after installing: `python predict.py` Then I get the following error: `RuntimeError: CUDA out of memory. Tried to allocate 1024.00 MiB (GPU...