Digital Africa Plantation Counting Challenge
Helping Côte d'Ivoire
$10 000 USD
Challenge completed over 2 years ago
Prediction
Computer Vision
Object Detection
701 joined
219 active
Start
Feb 23, 23
Close
Apr 09, 23
Reveal
Apr 09, 23
Notebook
Notebooks · 11 Mar 2023, 20:52 · 4
Hey every, how can solve this issue : CUDA out of memory. Tried to allocate 106.00 MiB (GPU 0; 15.90 GiB total capacity; 14.74 GiB already allocated; 55.75 MiB free; 14.97 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
1. Resize the image Height and width
2 reduce the batch_size for both training and validation dataloaders
these 2 should help
Sometimes reducing the neural network size / number of neurons (units) helps.
Thanks, i try it!
move loss to cpu
loss.to('cpu')
this may help i guess, i am not sure