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