原创 深度學習應該選擇tensorflow還是pytorch？一個場景分類任務惹出的問題

最近在做一個場景分類的任務，用AlexNet在Places數據集上做場景的分類訓練，使用的框架是TF，自己不到2G的顯存很容易就崩潰了，一直出現如下錯誤： ResourceExhaustedError (see above for tra

2018-08-22 00:12:35 18