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