keras之多GPU训练方法

在上一篇博客中,咱们利用keras框架训练yolov3,训练脚本默认采用的是一块GPU,因为咱们有多块GPU,所以能够设置多块GPU训练来加快训练速度。python 实现方法很简单,首先在头文件中添加如下内容网络 from keras.utils import multi_gpu_model 而后找到本身构建网络的地方,在我这里,第一次构建model是如下的语句框架 if is_tiny_vers
相关文章
相关标签/搜索