Caffe參數交換源碼分析

對境準備:對於多個GPU而言,一臺機器2個GPU,參數交換的流程圖:          參數交換從main()進入train()函數,在train函數中找到對應源碼爲: 1 . . . . . 2 if (gpus.size() > 1) { 3 caffe::P2PSync<float> sync(solver, NULL, solver->param()); 4 sync.r
相關文章
相關標籤/搜索