how to reduce memory cost for transfer learning with semantic segmentation

Question

Lukas Houben 2021-4-12

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/800196-how-to-reduce-memory-cost-for-transfer-learning-with-semantic-segmentation

回答： Srivardhan Gadila 2021-4-15

I am trying to use transfer learning with semantic segmentation to classify 960x720x3 images. I am using matlab on my surface with 8GB RAM and a build in GPU. When I try to run the code I get the message that the memory is not sufficient. To resolve the issue I thought about using PCA on my pixel data but I only found a comparable example that used parallel computing.

Therefore I would like to ask if there are any other solutions to reduce the memory cost of my code or if there is any way to get acces to more memory via matlab? (perhaps something like cloud computing)

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

请先登录，再进行评论。

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

Answer 1

Srivardhan Gadila 2021-4-15

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/800196-how-to-reduce-memory-cost-for-transfer-learning-with-semantic-segmentation#answer_675581

Try reducing the mini-batch size using the 'MiniBatchSize' option of trainingOptions.

If reducing the mini-batch size does not work, then try using a smaller network, reducing the number of layers, or reducing the number of parameters or filters in the layers.

If the gpu memory is still not sufficient then you can train the network on the cpu by using the 'ExecutionEnvironment' option of trainingOptions.

You can refer to Deep Learning in Parallel and in the Cloud & Deep Learning in the Cloud for information related to cloud computing.