Why training stops showing error (out of memory) during implementing transfer learning with pre-trained network despite having a laptop of well configuration?

Question

ZEBA 2024-4-22

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2109941-why-training-stops-showing-error-out-of-memory-during-implementing-transfer-learning-with-pre-trai

评论： ZEBA 2024-4-24

Trying to use trasnfer learning method for classsify images but I have failed to run. It always stop showing error: out of memory. I have faced the same problem when I have tried to build a large and complex cnn architecture from scratch. At first I have thought that too large dataset caused the problem.I have tried my model with only few images to test and even have tried to run in better configured pc than my laptop also, but I couln't get rid of the problem. Now I am facing the same issue while trying transfer learing. I do need both runinng a complex network from scratch and using pre-trained network.

I have follwed the documentation of transfer learning using the Prepare Network for Transfer Learning Using Deep Network Designer.

The details of images, choosen pre-trained methods and pc confuguration are:

images no: 60,000

class of image: 13

pre-trained network: DenseNet-201

laptop configuration: Ryzen 7, RAM 16GB, GPU NVIDIA GEFORCE RTX 3050

I am also attaching the Error msg showing and gpu details from gpuDevice:

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Joss Knight 2024-4-22

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2109941-why-training-stops-showing-error-out-of-memory-during-implementing-transfer-learning-with-pre-trai#answer_1445571

编辑：Joss Knight 2024-4-22

You have a very small GPU I'm afraid. You need to lower your MiniBatchSize until it works. Also, set your ExecutionEnvironment to auto since you only have one GPU.

If this doesn't work your only option is to use a smaller network or a bigger GPU.

You could just train on the CPU of course, if you don't mind it taking longer. At least that will likely have plenty of memory, and the way GPUs tend to be throttled on laptops it probably won't be much slower.