MATLAB Answers

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

3 views (last 30 days)
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 !


Sign in to comment.

Accepted Answer

Srivardhan Gadila
Srivardhan Gadila on 18 Aug 2020
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.


Sign in to comment.

More Answers (0)




Community Treasure Hunt

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

Start Hunting!