Matlab：沒有足夠的GPU內存用於分類

Question

我已經從頭開始用我自己的類別（現在只是花類型）培訓了Mathworks提供的Resnet50，並且想要測試它。

雖然在將訓練選項中的MiniBatchSize更改爲5而不是10後，訓練沒有問題，但分類不會啓動，因爲它表示沒有足夠的內存可用。我正在使用imageDataset中的測試圖像，就像在培訓期間一樣。

有沒有辦法強制CPU分類或調整一些選項，以便它能夠運行？ 歡迎任何其他想法！

Answer 1

我自己實際解決了這個問題：既分類和預測採用名稱 - 值對參數，如'MiniBatchSize'或'ExecutionEnvironment'。我能夠用這些東西解決我的問題，並修改值。