Yes I’ve tried it again but have got the following error
Must I decrease it more?

RuntimeError: CUDA out of memory. Tried to allocate 88.00 MiB (GPU 0; 4.00 GiB total capacity; 2.27 GiB already allocated; 10.45 MiB free; 2.36 GiB reserved in total by PyTorch)