keras之多GPU訓練方法

在上一篇博客中,咱們利用keras框架訓練yolov3,訓練腳本默認採用的是一塊GPU,因爲咱們有多塊GPU,所以能夠設置多塊GPU訓練來加快訓練速度。python 實現方法很簡單,首先在頭文件中添加如下內容網絡 from keras.utils import multi_gpu_model 而後找到本身構建網絡的地方,在我這裏,第一次構建model是如下的語句框架 if is_tiny_vers
相關文章
相關標籤/搜索