Faster R-CNN学習時のエラーについて

Faster R-CNNの学習時に以下のようなエラーが表示されます。

グラフィックボード：NVIDIA GeForce RTX 3060（メモリ容量12GB）

入力画像の解像度は500x500、MiniBatchSizeは8です。

この際の原因と解決策をご教授ください。

使い方によるエラー visiongpuROIMaxPoolingForward
デバイスで許容される最大変数サイズを超えました。

エラー: nnet.internal.cnn.layer.util.ROIMaxPooling2DGPUStrategy/forward (行 24)
            Z = visiongpuROIMaxPoolingForward(X, double(roiTransposed), gridSize(1), gridSize(2));

エラー: nnet.internal.cnn.layer.ROIMaxPooling2DLayer/forward (行 86)
                [Z, memory] = this.ExecutionStrategy.forward(X{1}, scaledBBoxes, this.GridSize);

エラー: nnet.internal.cnn.util.executeWithStagedGPUOOMRecovery (行 11)
        [ varargout{1:nOutputs} ] = computeFun(X1, X2);

エラー: nnet.internal.cnn.DAGNetwork/forwardPropagationWithMemory (行 379)
                [outputActivations, memory] = nnet.internal.cnn.util.executeWithStagedGPUOOMRecovery( ...

エラー: nnet.internal.cnn.DAGNetwork/computeGradientsForTraining (行 733)
                this.forwardPropagationWithMemory( X, ...

エラー: nnet.internal.cnn.Trainer/computeGradients (行 227)
            [gradients, predictions, states] = net.computeGradientsForTraining(X, Y, propagateState);

エラー: nnet.internal.cnn.Trainer/train (行 126)
                    [gradients, Y, states] = this.computeGradients(net, X, T);

エラー: vision.internal.cnn.trainNetwork (行 120)
trainedNet = trainer.train(trainedNet, trainingDispatcher);

エラー: trainFasterRCNNObjectDetector>iTrainEndToEnd (行 955)
    [net, info] = vision.internal.cnn.trainNetwork(...

エラー: trainFasterRCNNObjectDetector (行 462)
    [detector, info] = iTrainEndToEnd(trainingData, fastRCNN, options, params, executionSettings, imageInfo);

エラー: fasterRCNN_training_exe (行 58)
    [detector, info] = trainFasterRCNNObjectDetector(trainingData,lgraph,options, ...

Answer 1

[Hiro Yoshino]

https://jp.mathworks.com/matlabcentral/answers/372665-fastrcnn-out-of-memory

こちらの回答が参考になるかと思います。

[太一 宮柱]

回答いただきありがとうございます。

参考にさせていただきます。