How to use GPU on model that was imported from the "Deep Learning Toolbox Importer for TensorFlow-Keras Models" ?
2 visualizzazioni (ultimi 30 giorni)
Mostra commenti meno recenti
Tuong Nguyen Minh
il 15 Ago 2020
Risposto: Srivardhan Gadila
il 18 Ago 2020
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 Commenti
Risposta accettata
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.
0 Commenti
Più risposte (0)
Vedere anche
Categorie
Scopri di più su Deep Learning Toolbox in Help Center e File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!