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

2 visualizzazioni (ultimi 30 giorni)
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 !

Risposta accettata

Srivardhan Gadila
Srivardhan Gadila il 18 Ago 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.

Più risposte (0)

Categorie

Scopri di più su Deep Learning Toolbox in Help Center e File Exchange

Prodotti


Release

R2020a

Community Treasure Hunt

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

Start Hunting!

Translated by