命令行参数 ——tf.app.flags和python argparse
-
深度学习神经网络往往有过多的Hyperparameter需要调优,优化算法、学习率、卷积核尺寸等很多参数都需要不断调整,使用命令行参数是非常方便的。有两种实现方式,一是利用python的argparse包,二是调用tensorflow自带的app.flags实现。
-
tf.app.flags
import tensorflow as tf
flags = tf.app.flags
flags.DEFINE_string('path',
'data/test_data',
'Path to annotation')
FLAGS = flags.FLAGS
命令行: python test.py --path 'data'
- python argparse
import argparse
parser = argparse.ArgumentParser(description='')
parser.add_argument('--path', default='data/test_data')
args = parser.parse_args()
命令行: python test.py --path 'data'