cifar10_multi_gpu_train.py

所需积分/C币:0 2020-11-29 01:59:06 10KB PY
收藏 收藏
举报

TensorFlow多GPU训练官方示例,绝壁不是草泥马写个CUDA_VISIBLE_DEVICES就可以解决的事情。整体思路关键是求变量在各个GPU上的梯度平均值,然后用梯度平均值去更新变量。

...展开详情
下载并关注上传者 低至0.43元/次 身份认证VIP会员低至7折
抢沙发
一个资源只可评论一次,评论内容不能少于5个字
  • 签到新秀

    累计签到获取,不积跬步,无以至千里,继续坚持!
  • 分享学徒

    成功上传1个资源即可获取
关注 私信 TA的资源
上传资源赚积分or赚钱
最新推荐