2017-12-02 60 views
0

我已經從頭開始用我自己的類別(現在只是花類型)培訓了Mathworks提供的Resnet50,並且想要測試它。Matlab:沒有足夠的GPU內存用於分類

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

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

回答

0

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

相關問題