tensorflow中flags的用法
from absl import app, flags, logging
之前看MAML算法时就遇到大量的flags类,现在看fswf又有很多,特意查了一下相关资料,其用法是
flags可以帮助我们通过命令行来动态的更改代码中的参数。
一个demo(节选):
from absl import app, flags, logging
flags.DEFINE_string('type', '','input type.')
flags.DEFINE_integer('index', 0,'input idnex')
FLAGS = flags.FLAGS
print(FLAGS.type)
print(FLAGS.index)
在命令行输入
python test.py --index=0 --type=ps
结果:
ps
0
修改的话在命令行输入对应代码,不输入代表使用默认的