CUDA out of memory error

[jmhjimenez]

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?