美文网首页
tensorflow命令行参数

tensorflow命令行参数

作者: 空谷幽兰_c411 | 来源:发表于2020-03-12 11:52 被阅读0次

    一个使用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函数

    相关文章

      网友评论

          本文标题:tensorflow命令行参数

          本文链接:https://www.haomeiwen.com/subject/yfoyjhtx.html