python tensorflow 命令行输入

tensorflow 命令行输入有两种：

  1. 利用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
  1. 利用tensorflow的tf.flags组件
FLAGS = tf.flags.FLAGS

tf.flags.DEFINE_integer('batch_size', 1, 'batch size, default: 1')

batch_size = FLAGS.batch_size
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容