我在matlab做图像分类的时候遇到 classify函数出现GPU内存不足的问题,问题如下图:
我的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{:});