在编写python程序中,可以使用argparse解析命令行参数来传递参数,进而根据需要控制程序的运行方式。
例子:
import argparse # create an ArgumentParser object parser = argparse.ArgumentParser(description="test a parser") # add arguments parser.add_argument("-b", "--batch_size", type=int, default=16, help="batch size.") parser.add_argument("-g", "-gpu_ids", type=str, default="0,1", help="the visible gpu device ids.") parser.add_argument("-c", "--config", type=str, default="training0901/config.py", help="config file.") # get the args object args = parser.parse_args() # use the arguments batch_size = args.batch_size gpu_ids = args.gpu_ids config = args.config