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 !

  0 Comments

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.

  0 Comments

Sign in to comment.

More Answers (0)

Products


Release

R2020a

Community Treasure Hunt

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

Start Hunting!