一个使用python argparse库,一个使用tf.flags,本质是基于argparse再封装
tf定义了tf.app.flags用于支持接受命令行参数,包括接受字符串,整型,浮点型,布尔型参数.可以通过命令行接受参数或者设置默认值
import tensorflow as tf
flags = tf.flags
FLAGS = flags.FLAGS
flags.DEFINE_string('data_dir', '../data/mnist', help='input_data')
flags.DEFINE_float('learning_rate', 1e-4, help='learning rate')
flags.DEFINE_integer('epoch', 10, help='epoch')
def main(_):
pass
if __name__ == '__main__':
flags.mark_flag_as_required('save_dir')
tf.compat.v1.app.run()
tf.app.run()作用是解析命令行参数,调用main函数
网友评论