How to use GPU on model that was imported from the "Deep Learning Toolbox Importer for TensorFlow-Keras Models" ?

I have trained a network model in Keras and has successfully been able to import such model into Mathlab 2020a. How ever I am not sure how to use GPU for that model to accelerate the speed ?

Please help me thank you !

Answer 1

[Srivardhan Gadila]

You can use the following Name-Value pair arguments: ExecutionEnvironment & Acceleration of the predict function in order to use the GPU for prediction.

Make sure that you are using a SeriesNetwork or a DAGNetwork object with the predict function. If you have used importKerasNetwork then no issues. If you have used importKerasLayers then convert the layers/layerGraph to a SeriesNetwork or a DAGNetwork object using the assembleNetwork function.