質問: alexnetの転移学習をgpu環境で行うには 以下のリンクを参考にalexnetの転移学習をgpuで行いたいと考えております。 どのようなgpuの追加コードを加えると動くでしょうか。 https://jp.mathworks.com/help/deeplearning/examples/transfer-learning-using-alexnet.html どうぞよろしくお願いいたします。
回答: ドキュメント内の trainnetwork 関数でニューラルネットワークのパラメタ学習を行っています。trainnetwork 関数は、デフォルトで gpu を使用する仕様になっています。そのため、gpu を使用するためのコードを追加する必要はありません。cpu を使用したい場合は 'executionenvironment' オプションを 'cpu' にします。 もし gpu のあるマシンで cpu でしか実行されていない場合は、gpudevice を使用して、matlab が gpu を認識できているかを確認することができます。