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
相关文章
相关标签/搜索