如何解決GPU太小
2019/07/26 下午 03:19
機器學習共學討論版
黃關明
觀看數:78
回答數:3
收藏數:0
想問一下,如果神經網路太深,GPU不夠算的時候該怎麼辦
回答列表
-
2019/07/26 下午 04:34張維元 (WeiYuan)贊同數:2不贊同數:1留言數:1
買比較強的 GPU 嗎?或是直接租雲端服務?類似:https://aws.amazon.com/tw/ec2/instance-types/p3/
-
2019/07/26 下午 11:28Jimmy贊同數:3不贊同數:0留言數:2
Hi 關明!
通常來說,在怎麼深的網路,目前的 GPU 應該都能夠至少一次處理數張影像,所以通常是
1. 你的影像超級大 (1000 x 1000 pixels 以上)
2. 你一次訓練的 batch size 太大!
造成 GPU 的記憶體不夠處理,先試著調整這兩個東西,通常在深的網路應該都還是有辦法訓練的!
-
2019/07/27 下午 11:12astronomer贊同數:0不贊同數:0留言數:2
記得有些 framework (如 pytorch) 可以累積多次 batch 再一次算 backward propagation, 這樣可以解決 GPU 計算太小, 卻又不想降低 batch size
就不知道 keras 要怎麼設了