别裁伪体亲风雅,转益多师是汝师。这篇文章主要讲述tensorflow API _ 2 (tf.app.flags.FLAGS)相关的知识,希望能为你提供帮助。
tf.app.flags.FLAGS 的使用,主要是在用命令行执行程序时,需要传些参数,代码如下:
新建一个名为:app_flags.py 的文件。
#coding:utf-8
import tensorflow as tf
FLAGS = tf.app.flags.FLAGS
tf.app.flags.DEFINE_string("train_data_path", "/home/libo3/train.txt", "training data dir")
tf.app.flags.DEFINE_string("log_dir", "./logs", " the log dir")
tf.app.flags.DEFINE_integer("max_sentence_len", 80, "max num of tokens per query")
tf.app.flags.DEFINE_integer("embedding_size", 50, "embedding size")
tf.app.flags.DEFINE_float("learning_rate", 0.001, "learning rate")
def main(unused_argv):
train_data_path = FLAGS.train_data_path
print("train_data_path", train_data_path)
max_sentence_len = FLAGS.max_sentence_len
print("max_sentence_len", max_sentence_len)
embdeeing_size = FLAGS.embedding_size
print("embedding_size", embdeeing_size)
abc = tf.add(max_sentence_len, embdeeing_size)
init = tf.global_variables_initializer()
#with tf.Session() as sess:
#sess.run(init)
#print("abc", sess.run(abc))
sv = tf.train.Supervisor(logdir=FLAGS.log_dir, init_op=init)
with sv.managed_session() as sess:
print("abc:", sess.run(abc))
# sv.saver.save(sess, "/home/yongcai/tmp/")
# 使用这种方式保证了,如果此文件被其他文件 import的时候,不会执行main 函数
if __name__ == ‘__main__‘:
tf.app.run()
# 解析命令行参数,调用main 函数 main(sys.argv)
调用方法:
其中参数可以根据需求进行修改。
- python app_flags.py --train_data_path < 绝对路径 train.txt> --max_sentence_len 100 --embedding_size 100 --learning_rate 0.05
- python app_flags.py
【tensorflow API _ 2 (tf.app.flags.FLAGS)】
推荐阅读
- Android 开发中Service完全解析
- android -------- Eclipse下的NDK配置环境
- 背单词app测试
- 背单词类APP测试与评估
- 如何更改 Apple ID 国家或地区
- Android智能手机中各种音频场景下的audio data path
- [Android Security]APK自我保护 - 字符串处理
- 「部署日记」Android Studio乱码解决方案
- android编程测试