Alexnetの転移学習をGPU環境で行うには

以下のリンクを参考にAlexnetの転移学習をGPUで行いたいと考えております。

どのようなGPUの追加コードを加えると動くでしょうか。

https://jp.mathworks.com/help/deeplearning/examples/transfer-learning-using-alexnet.html

どうぞよろしくお願いいたします。

Answer 1

[mizu]

ドキュメント内の trainNetwork 関数でニューラルネットワークのパラメタ学習を行っています。trainNetwork 関数は、デフォルトで GPU を使用する仕様になっています。そのため、GPU を使用するためのコードを追加する必要はありません。CPU を使用したい場合は 'ExecutionEnvironment' オプションを 'cpu' にします。

もし GPU のあるマシンで CPU でしか実行されていない場合は、gpuDevice を使用して、MATLAB が GPU を認識できているかを確認することができます。

[ssk]

デフォルトでGPUを使用する仕様となっている旨、ご教示いただきありがとうございました。