RuntimeError: CUDA out of memory. Tried to allocate 970.00 MiB (GPU 0; 7.79 GiB total capacity; 4.39 GiB already allocated; 57.06 MiB free; 5.63 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

[bling-bling-only]

请问有什么可以避免显卡爆显存的问题，batch size默认为1，然后图片切割尺寸也调小过，在--scale 2000的情况下只能训练20个epoch，且无法验证