Python之FLAGS用法（python的flag）

Python之FLAGS用法（python的flag）

from absl import appfrom absl import flagsFLAGS = flags.FLAGSflags.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：​​https://huangshiyu13.github.io/​​

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。