Why I running into GPU out of memory when using Segnet?
2 次查看(过去 30 天)
显示 更早的评论
The segnet works well when we input 3000 224*224 RGB images(1800 for training,900 for validation and 300 for testing). However, when we input more than 3000 224*224 RGB images, the GPU out of memory from the start of the training process, even the value of MiniBatchSize is 1. My GPU is 8GB GTX1080, and the screenshot is shown as follows.
Thank you very much for the help and looking forward to the reply!
0 个评论
回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!