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

10 次查看(过去 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
"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