tf.app.flags说明

很多时候在运行python代码的时候我们需要从外部定义参数,从而避免每次都需要改动代码,tf就提供了tf.app.flags功能。

import tensorflow as tf
# 学习使用 tf.app.flags 使用,全局变量
# tf.app.flags.DEFINE_string("param_name", "default_val", "description")
tf.app.flags.DEFINE_boolean("clean",       False,      "clean train folder")
tf.app.flags.DEFINE_integer("batch_size",  8,        "batch size")
tf.app.flags.DEFINE_float("dropout",       0.5,        "Dropout rate")
tf.app.flags.DEFINE_string("log_dir", "./logs", " the log dir")
FLAGS=tf.app.flags.FLAGS
print(FLAGS.dropout)

 

posted @ 2019-11-24 17:58  光彩照人  阅读(254)  评论(0编辑  收藏  举报