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

Question

0 votes

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, ...