Digital Africa Plantation Counting Challenge
Can you create a semi-supervised algorithm to count trees in plantations in Côte d'Ivoire?
Prize
$10 000 USD
Time
20 days to go
Participants
153 active · 460 enrolled
Helping
Côte dIvoire
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