1. 首页
  2. 服务器应用
  3. Proxy
  4. cifar10_multi_gpu_train.py

cifar10_multi_gpu_train.py

上传者: 2020-11-29 17:14:21上传 PY文件 10.4KB 热度 8次
TensorFlow多GPU训练官方示例,绝壁不是草泥马写个CUDA_VISIBLE_DEVICES就可以解决的事情。整体思路关键是求变量在各个GPU上的梯度平均值,然后用梯度平均值去更新变量。
用户评论