Matlab：分類にGPUメモリが不十分です

Question

Mathworksが提供するResnet50を独自のカテゴリ（今のところ花のタイプ）でゼロから訓練し、テストしたかったのです。

trainingOptionsでMiniBatchSizeを10ではなく5に変更しても、トレーニングは問題なく動作しましたが、使用可能なメモリが不足しているため、分類が開始されません。私は、トレーニング中のようにimageDatasetでテスト画像を使用しています。

CPU分類を強制したり、いくつかのオプションを調整して実行する方法はありますか？ 他のアイデアも大歓迎です！

Answer 1

私は実際に自分自身を解決しました。「MiniBatchSize」や「ExecutionEnvironment」などの名前 - 値ペア引数を分類して予測します。私はこれらを使って自分の問題を解決し、その値を微調整することができました。