LiliiiH

Thank you so much!

I met this error when I wanted to run train.py: RuntimeError: CUDA out of memory. Tried to allocate 400.00 MiB (GPU 0; 24.00 GiB total capacity; 18.64 GiB already allocated;...