热门搜索 :
考研考公
您的当前位置:首页正文

TensorFlow使用flags自定义命令行参数

来源:东饰资讯网

简介

TensorFlow项目例子中经常出现tf.app.flags,它支持应用从命令行接受参数,可以用来指定集群配置等。

注意,目前TensorFlow的flags是基于gflag实现的,由于稳定版不一定使用gflag,因此官方文档并没有相关介绍,等API确定后才有官方文档。

使用实例

下面是最简单的tf.flags例子,可以指定多个参数和不同默认值。

如果不传参数,可以使用默认参数,传入参数后输出结果如下。

Top