tensorflow app

发表于 | 分类于 | 评论数：

tf.app.flags

代码示例

代码地址

1
2
3
4
5
6
7
8
9
10
11
12
13
14
flags.py
import tensorflow as tf

flags = tf.app.flags
flags.DEFINE_string('model', 'mxx', 'Type of model')
flags.DEFINE_boolean('gpu','True', 'use gpu?')
FLAGS = flags.FLAGS

def main(_):
    for k,v in FLAGS.flag_values_dict().items():
        print(k, v)

if __name__ == "__main__":
    tf.app.run(main)

传递参数的方法有两种，一种是命令行~$:python flags.py --model hhhh ，一种是pycharm中传递参数。