python argparse True False(action="store_true")

argparser를 활용한 True / False 반환 하기 : action="store_true"

 

use_GPU 변수에 True 혹은 False 를 담는 argparse 코드는 아래와 같습니다.

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--use_GPU", action="store_true") # use_GPU를 사용하면 true를 저장한다로 해석합니다.
args = parser.parse_args()

if args.use_GPU == False:
    print(args.use_GPU)
else:
    print(args.use_GPU)

 

실행 방법은 아래와 같습니다.

 # 터미널에 아래와 같이 입려하면 use_GPU 인자를 사용하지 않으면 False를 반환합니다.
 # 인자 작성 안함
 python action_test.py

# use_GPU 인자를 사용하면 True를 반환하려면 --use_GPU 인자를 작성합니다.
python action_test.py --use_GPU