TensorFlow 中 tf.app.flags.FLAGS 的用法介绍

转载自：https://blog.csdn.net/lyc_yongcai/article/details/73456960

下面介绍 tf.app.flags.FLAGS 的使用，主要是在用命令行执行程序时，需要传些参数，代码如下：

新建一个名为：app_flags.py 的文件。

#coding:utf-8# 学习使用 tf.app.flags 使用，全局变量
# 可以再命令行中运行也是比较方便，如果只写 python app_flags.py 则代码运行时默认程序里面设置的默认设置
# 若 python app_flags.py --train_data_path <绝对路径 train.txt> --max_sentence_len 100
#    --embedding_size 100 --learning_rate 0.05  代码再执行的时候将会按照上面的参数来运行程序import tensorflow as tfFLAGS = tf.app.flags.FLAGS# tf.app.flags.DEFINE_string("param_name", "default_val", "description")
tf.app.flags.DEFINE_string("train_data_path", "/home/yongcai/chinese_fenci/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_pathprint("train_data_path", train_data_path)max_sentence_len = FLAGS.max_sentence_lenprint("max_sentence_len", max_sentence_len)embdeeing_size = FLAGS.embedding_sizeprint("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

则会执行程序时会自动调用程序中 default 中的参数。

TensorFlow 中 tf.app.flags.FLAGS 的用法介绍相关推荐

python中字符串函数的用法_python中字符串内置函数的用法介绍（代码）
本篇文章给大家带来的内容是关于python中字符串内置函数的用法介绍(代码) ,有一定的参考价值,有需要的朋友可以参考一下,希望对你有所帮助. capitalize() 首字母大写a='somewor ...
tensorflow代码中tf.app.run()什么意思
# 前面的代码省略了... ... ... ... def main(argv=None):mnist = input_data.read_data_sets("F:\mydata\Tens ...
TensorFlow中 tf.space_to_depth()函数的用法
目录一.函数定义二.解释范例三.代码验证一.函数定义通俗易懂些,就是把输入为[batch, height, width, channels]形式的Tensor,其在height和width维 ...
Tensorflow中tf.ConfigProto()详解
参考Tensorflow Machine Leanrning Cookbook tf.ConfigProto()主要的作用是配置tf.Session的运算方式,比如gpu运算或者cpu运算具体代码如 ...
tensorflow中tf.gradients()解析
tf.gradients()解析及grad_ys在xs为(?, 1)时的理解问题简介使用tensorflow1.15学习时,有一项tf.gradients的代码,其中用到了grad_ys这个参数, ...
tensorflow中tf.nn.xw_plus_b
tf.nn.xw_plus_b((x, weights) + biases) 相当于tf.matmul(x, weights) + biases #-*-coding:utf8-*- import t ...
mysql中模糊查询的四种用法介绍
下面介绍mysql中模糊查询的四种用法: 1,%:表示任意0个或多个字符.可匹配任意类型和长度的字符,有些情况下若是中文,请使用两个百分号(%%)表示. 比如 SELECT * FROM [user ...
mysql 中模糊查询的四种用法介绍
下面介绍mysql中模糊查询的四种用法: 1,%:表示任意0个或多个字符.可匹配任意类型和长度的字符,有些情况下若是中文,请使用两个百分号(%%)表示. 比如 SELECT * FROM [user] ...
html中ul和li标签的用法介绍
普通的显示数据的时候,ul就是项目列表,li就是列表项,可以用来显示数据,如果用于DIV+CSS布局的话,ul+li可以替换表格的作用,具体的设置,如果宽度高度.行间距.背景边框需要配合CSS一起设置 ...

TensorFlow 中 tf.app.flags.FLAGS 的用法介绍

TensorFlow 中 tf.app.flags.FLAGS 的用法介绍相关推荐

最新文章

热门文章