Python中如何正确使用FLAGS参数？

本文共计137个文字，预计阅读时间需要1分钟。

pythonfrom absl import appfrom absl import flagsFLAGS=flags.FLAGSflags.DEFINE_string('gpu', None, 'Comma-separated list of GPUs to use.')

def main(argv): del argv if FLAGS.gpu: os.environ['CUDA_VISIBLE_DEVICES']=FLAGS.gpu else: print('Please specify GPU(s).')

from absl import app
from absl import flags

FLAGS = flags.FLAGS

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

def main(argv):
del argv
if FLAGS.gpu:
os.environ['CUDA_VISIBLE_DEVICES'] = FLAGS.gpu
else:
print('Please assign GPUs.')
exit()

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

python test.py --gpu 0

黄世宇/Shiyu Huang's Personal Page：​​huangshiyu13.github.io/​​