Alexnetの転移学習をGPU環境で行うには
1 visualización (últimos 30 días)
Mostrar comentarios más antiguos
以下のリンクを参考にAlexnetの転移学習をGPUで行いたいと考えております。
どのようなGPUの追加コードを加えると動くでしょうか。
どうぞよろしくお願いいたします。
0 comentarios
Respuesta aceptada
mizu
el 9 de Mzo. de 2019
ドキュメント内の trainNetwork 関数でニューラルネットワークのパラメタ学習を行っています。trainNetwork 関数は、デフォルトで GPU を使用する仕様になっています。そのため、GPU を使用するためのコードを追加する必要はありません。CPU を使用したい場合は 'ExecutionEnvironment' オプションを 'cpu' にします。
Más respuestas (0)
Ver también
Categorías
Más información sobre Deep Learning Toolbox en Help Center y File Exchange.
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!