Closed wowwowya closed 10 months ago
it shows torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 256.00 MiB (GPU 0; 8.00 GiB total capacity; 933.87 MiB already allocated; 4.95 GiB free; 948.00 MiB 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
Hi, training ITS needs about 17G memory.
How many epochs are needed in total for training on ITS?thankyou
Hi, the hyper-parameters we used are shown in the main.py.
How much GPU memory do you have? I'm training a model on reside-indoor, and it's showing 'CUDA out of memory' error. My graphics card is 4060 with 8GB.