[TensorFlow] tf.app

tf.flags

公式ドキュメント: tf.flags

argparse をラップしたコマンドライン引数を解析するためのモジュール。
公式のサンプルなどで利用されているのを見かける。

引数の型によって、以下の4種類がある。
tf.flags.DEFINE_string(flag_name, default_value, docstring)
tf.flags.DEFINE_integer(flag_name, default_value, docstring)
tf.flags.DEFINE_boolean(flag_name, default_value, docstring)
DEFINE_float(flag_name, default_value, docstring)

例:

import tensorflow as tf
tf.flags.DEFINE_string("msg", "hello", "Message to print.")
tf.flags.DEFINE_string("num", "0", "Number to print.")
FLAGS = tf.flags.FLAGS

print(FLAGS.msg)
print(FLAGS.num)
$ python test.py --msg hoge
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcudnn.so.5 locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcufft.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.so.8.0 locally
hoge
0

コメントを残す

メールアドレスが公開されることはありません。