tensorflow接受命令行参数

编程语言 > 机器学习 tensorflow python

发布时间 : 2020-07-14 00:00

字数:118 阅读 :

# -*- coding=utf-8 -*-

import tensorflow  as tf

FLAGS = tf.app.flags.FLAGS

tf.app.flags.DEFINE_string('ckpt_path', 'model/model.ckpt-100000', '''模型保存路径''')
tf.app.flags.DEFINE_float('learning_rate', 0.0001, '''初始学习率''')
tf.app.flags.DEFINE_integer('train_steps', 50000, '''总的训练轮数''')
tf.app.flags.DEFINE_boolean('is_use_gpu', False, '''是否使用GPU''')

print('模型保存路径： {}'.format(FLAGS.ckpt_path))
print('初始学习率： {}'.format(FLAGS.learning_rate))
print('总的训练次数： {}'.format(FLAGS.train_steps))



# tmp python tensorflow接受命令行参数.py --ckpt_path /to/my/path --learning_rate 0.000001 --train_steps 10000 --is_use_gpu False

转载请注明来源 https://tianweiye.github.io