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
Advanced
Prediction
Computer Vision
Object Detection
Agriculture
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
Discussion 4 answers

1. Resize the image Height and width

2 reduce the batch_size for both training and validation dataloaders

these 2 should help

11 Mar 2023, 20:55
Upvotes 2

Sometimes reducing the neural network size / number of neurons (units) helps.

11 Mar 2023, 21:33
Upvotes 2

Thanks, i try it!

12 Mar 2023, 14:43
Upvotes 0

move loss to cpu

loss.to('cpu')

this may help i guess, i am not sure

18 Mar 2023, 05:06
Upvotes 0