python基础--absl.flags

之前在tensorflow的mnist例程中看到了使用 absl.flags的方法来载入和解析参数的，出于学习的目的，就自己试验了一下，

代码如下:

# *_*coding:utf-8 *_*
# athor:auto

import sys, os
from absl import app
from absl import flags
from official.utils.flags import core as flags_core


FLAGS = flags.FLAGS
flags.DEFINE_string('gpu', None, 'comma separated list of GPU to use.')


def flagtest(argv):
    del argv
    if FLAGS.gpu:
        print("gpu is %s" % FLAGS.gpu)
        os.environ['CUDA_VISIBLE_DEVICES'] = FLAGS.gpu
    else:
        print('Please assign GPUs.')
        exit()

def main(argv):
    flags_core.define_base()
    flags_core.define_performance(num_parallel_calls=False)
    flags_core.define_image()
    flags.adopt_module_key_flags(flags_core)

if __name__ == '__main__':
    app.run(flagtest)

其中main中的几个调用都是源自于tensorflow的model/official，里面的函数大多是model/official/utils/flags/core.py内定义好的一些默认参数。
在mnist例子中还可以这样添加自定义项：

  flags_core.set_defaults(data_dir='./tmp/mnist_data',
                          model_dir='./tmp/mnist_model',
                          batch_size=100,
                          train_epochs=40,
                          stop_threshold=0.998)

　　

参考:

https://blog.csdn.net/faith_binyang/article/details/80551941