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