Out of memory error

4 次查看（过去 30 天）

Sania Gul 2019-12-31

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/498541-out-of-memory-error

评论： Joss Knight 2022-6-12

I am trainning Unet. I have 7500 images and their masks. The size of individual image and mask is (512,180). I have divided the data for training and testing in ratio of 0.9:0.1. My training goes fine but i get Out of memory error after it. The network then doesnot perform any testing. Kindly tell me the solution

The error is given as.

Out of memory on device. To view more detail about available memory on the GPU, use 'gpuDevice()'. If the problem persists, reset the GPU by
calling 'gpuDevice(1)'.
Error in nnet.internal.cnn.layer.util.BatchNormalizationGPUStrategy/forwardPredict (line 14)
            Z = nnet.internal.cnngpu.batchNormalizationForwardPredict(X, beta, gamma, epsilon, inputMean, inputVar, channelDim);
Error in nnet.internal.cnn.layer.BatchNormalization/predict (line 159)
            Z = this.ExecutionStrategy.forwardPredict( ...
Error in nnet.internal.cnn.DAGNetwork/activations (line 571)
                    outputActivations = thisLayer.predict(XForThisLayer);
Error in DAGNetwork/calculateActivations (line 86)
        YBatch = predictNetwork.activations({X}, layerIndex, layerOutputIndex);
Error in DAGNetwork/activations (line 138)
Y = this.calculateActivations(X, layerIndex, layerOutputIndex, varargin{:});
Error in semanticseg>iClassifyImagePixels (line 447)
    allScores = activations(net, X, name, ...
Error in semanticseg>iProcessImageDatastoreSerially (line 663)
        L = iClassifyImagePixels(X, net, params);
Error in semanticseg (line 211)
        filenames = iProcessImageDatastoreSerially(imds, net, params);
Error in Step5NeuralNetwork (line 90)
    pxdsResults = semanticseg(imdsTest,net,'WriteLocation',tempdir,'Verbose',false);
 

4 个评论
显示 2更早的评论隐藏 2更早的评论

Anay Naik 2022-6-8

How do you change the miniBatchSize of a pretrained network? Is there a function? If so, how is it written and where should it be placed in the code?

Thank you!

Joss Knight 2022-6-12

You read the documentation and see that for a DAGNetwork, the predict, classify and activations functions have a MiniBatchSize option. Is that what you're using?

请先登录，再进行评论。

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