[学习博] tf.app.flags

tf.appp.flags

主要是在用命令行执行程序时，需要传些参数，其实也就可以理解成对argparse库进行的封装。

# coding:utf-8
# authors:wxl
# 学习tf.app.flags使用, 在命令行模式下运行更加方便

import tensorflow as tf

FLAGS = tf.app.flags.FLAGS
# 格式：tf.app.flags.DEFINE_string("param_name", "default_val", "description")
tf.app.flags.DEFINE_string("train_data_path", "/desktop/train.txt", "train data dir")
tf.app.flags.DEFINE_integer("epochs", 50, "epochs")
tf.app.flags.DEFINE_float("lr", 0.01, "learning_rate")

def main():
    train_data_path = FLAGS.train_data_path
    print("train_data_path",train_data_path)
    print("- " *30)
    epoch = FLAGS.epochs
    print("epoch", epoch)

    init = tf.global_variables_initializer()

    with tf.Session() as sess:
        sess.run(init)


if __name__ == '__main__':
    #该行代码表示如果当前是从其它模块调用的该模块程序，则不会运行main函数！而如果就是直接运行的该模块程序，则会运行main函数。
    tf.app.run()

调用1：

python 0618.py

2

python 0618.py  --train_data_path  ./test.py  --epochs 100