Hi, when tried this over Colab, I'm encountering the following error with a batch size of 2
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 11.17 GiB total capacity; 10.76 GiB already allocated; 12.81 MiB free; 101.88 MiB cached)
In call to configurable 'train' (<function train at 0x7fb2fb7f1e60>)
There's no error when the batch size is 1. Is there a way to fix this or do I really need a GPU with a higher memory??
For a GPU with 12Gb memory, is the batch size limited to only 1?
Hi, when tried this over Colab, I'm encountering the following error with a batch size of 2
There's no error when the batch size is 1. Is there a way to fix this or do I really need a GPU with a higher memory??
For a GPU with 12Gb memory, is the batch size limited to only 1?