Why does "predict" result in an error when setting the parameter "ExecutionEnvironment" in Deep Learning Toolbox?

조회 수: 9 (최근 30일)
I encountered an error when trying to perform inference with the help of GPU using a "dlnetwork" object.
I have a "dlnetwork" that I use with the "predict" function. It works as expected when I use the "predict" function alone:
>> allQval=predict(net,stateValue);
But specifying a value for "ExecutionEnvironment" explicitly triggers the error:
>> allQval=predict(net,stateValue,"ExecutionEnvironment","parallel");Error using dlnetwork/validateForwardInputs Incorrect number of network inputs. Network has 1 inputs, but 3 inputs were passed. To determine number and order of inputs, check network 'InputNames' property.
How can I use my GPU with the "predict" function?

채택된 답변

MathWorks Support Team
MathWorks Support Team 2024년 10월 28일
편집: MathWorks Support Team 2024년 10월 28일
"ExecutionEnvironment" is not a valid input argument for the "dlnetwork" "predict" function.
To use a GPU with the "predict" function, you can pass the input as "GPUArrays".
Alternatively, for batched-predictions, use the function "minibatchpredict" from R2024a. This uses the GPU through the "ExecutionEnvironment" argument.
Please note, the "Acceleration" input for "predict" is not the same as the "ExecutionEnvironment". "Acceleration"="mex" compiles and executes a "mex" function for inference using GPU Coder. For more information, please see the documentation: https://uk.mathworks.com/help/releases/R2024b/deeplearning/ref/dlnetwork.predict.html#mw_55eba9d5-3e06-467e-a6cd-e1cffe0569fd

추가 답변 (0개)

카테고리

Help CenterFile Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by