RuntimeError: CUDA out of memory in training

[Mabozaid0X1]

I follow all steps but when I run python train.py I find this error RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 4.00 GiB total capacity; 2.57 GiB already allocated; 74.77 MiB free; 2.85 GiB reserved in total by PyTorch) (malloc at ..\c10\cuda\CUDACachingAllocator.cpp:289) (no backtrace available)

[KainingYing]

maybe you should reduce the batchsize or use a bigger gpu memory machine