tensorflow 命令行输入有两种：
- 利用python自带库argparse
import argparse
import sys
def get_arguments():
parser = argparse.ArgumentParser(description='example network')
parser.add_argument('--batch_size', type=int,default=BATCH_SIZE,
help='How many files to process at once. Default:'+str(BATCH_SIZE)+'.')
return parser.parse_args()
args = get_arguments()
batch_size = args.batch_size
- 利用tensorflow的tf.flags组件
FLAGS = tf.flags.FLAGS
tf.flags.DEFINE_integer('batch_size', 1, 'batch size, default: 1')
batch_size = FLAGS.batch_size
网友评论