common.py 文件源码

python
阅读 32 收藏 0 点赞 0 评论 0

项目:iaf 作者: openai 项目源码 文件源码
def assign_to_gpu(gpu=0, ps_dev="/device:CPU:0"):
    def _assign(op):
        node_def = op if isinstance(op, tf.NodeDef) else op.node_def
        if node_def.op == "Variable":
            return ps_dev
        else:
            return "/gpu:%d" % gpu
    return _assign
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号