How much GPU memory do you have?

c-yn / FocalNet

[ICCV23] Focal Network for Image Restoration

MIT License

52 stars 3 forks source link

How much GPU memory do you have? #3

Closed wowwowya closed 10 months ago

wowwowya commented 11 months ago

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.

wowwowya commented 11 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

c-yn commented 10 months ago

Hi, training ITS needs about 17G memory.

wowwowya commented 10 months ago

How many epochs are needed in total for training on ITS?thankyou

c-yn commented 10 months ago

Hi, the hyper-parameters we used are shown in the main.py.