Mslihua
2021-09-01 15:16
采纳率: 100%
浏览 52

matlab上Classify函数报错问题。

我在matlab做图像分类的时候遇到 classify函数出现GPU内存不足的问题,问题如下图:

img

我的Textdata包含五个子文件夹,也就是分五类,然后每个文件夹里面有57张图片;尝试了去调节minbatchsize,但是好像不行,请问如何能解决这样的问题。谢谢!!

错误使用 nnet.internal.cnngpu.convolveBiasReluForward2D
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)'.

出错 nnet.internal.cnn.layer.util.ConvolutionReLUGPUStrategy/forward (line 20)
Z = nnet.internal.cnngpu.convolveBiasReluForward2D( ...

出错 nnet.internal.cnn.layer.ConvolutionReLU/predict (line 144)
Z = this.ExecutionStrategy.forward( X, ...

出错 nnet.internal.cnn.DAGNetwork/predict (line 421)
outputActivations = thisLayer.predict(XForThisLayer);

出错 DAGNetwork/calculatePredict>predictBatch (line 240)
YBatch = predictNetwork.predict(X);

出错 DAGNetwork/calculatePredict (line 17)
Y = predictBatch( ...

出错 DAGNetwork/classify (line 140)
scores = this.calculatePredict( ...

出错 SeriesNetwork/classify (line 505)
[labels, scores] = this.UnderlyingDAGNetwork.classify(X, varargin{:});

  • 收藏

1条回答 默认 最新

  • joel_1993 2021-09-01 15:59
    已采纳

    看不清楚你的截图,请弄清晰的问题截图

    已采纳该答案
    打赏 评论

相关推荐 更多相似问题