Alexnetの転移学習をGPU環境で行うには
显示 更早的评论
以下のリンクを参考にAlexnetの転移学習をGPUで行いたいと考えております。
どのようなGPUの追加コードを加えると動くでしょうか。
どうぞよろしくお願いいたします。
采纳的回答
mizu
2019-3-9
2 个投票
ドキュメント内の trainNetwork 関数でニューラルネットワークのパラメタ学習を行っています。trainNetwork 関数は、デフォルトで GPU を使用する仕様になっています。そのため、GPU を使用するためのコードを追加する必要はありません。CPU を使用したい場合は 'ExecutionEnvironment' オプションを 'cpu' にします。
更多回答（0 个）
类别
在 帮助中心 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息
