def parse_args(): parser = cli.ArgsParser() parser.add_argument( "--weight_type", default='pretrain', type=str, help= "Loading Checkpoints only support 'pretrain', 'finetune', 'resume'.") parser.add_argument("--eval", action='store_true', default=False, help="Whether to perform evaluation in train") parser.add_argument("--slim_config", default=None, type=str, help="Configuration file of slim method.") parser.add_argument("--enable_ce", type=bool, default=False, help="If set True, enable continuous evaluation job." "This flag is only used for internal test.") parser.add_argument("--fp16", action='store_true', default=False, help="Enable mixed precision training.") parser.add_argument("--fleet", action='store_true', default=False, help="Use fleet or not") args = parser.parse_args() return args
def parse_args(): parser = cli.ArgsParser() parser.add_argument("--eval", action='store_true', default=False, help="Whether to perform evaluation in train") parser.add_argument("-r", "--resume", default=None, help="weights path for resume") parser.add_argument("--slim_config", default=None, type=str, help="Configuration file of slim method.") parser.add_argument("--enable_ce", type=bool, default=False, help="If set True, enable continuous evaluation job." "This flag is only used for internal test.") parser.add_argument("--amp", action='store_true', default=False, help="Enable auto mixed precision training.") parser.add_argument("--fleet", action='store_true', default=False, help="Use fleet or not") parser.add_argument("--use_vdl", type=bool, default=False, help="whether to record the data to VisualDL.") parser.add_argument('--vdl_log_dir', type=str, default="vdl_log_dir/scalar", help='VisualDL logging directory for scalar.') parser.add_argument('--save_prediction_only', action='store_true', default=False, help='Whether to save the evaluation results only') parser.add_argument('--profiler_options', type=str, default=None, help="The option of profiler, which should be in " "format \"key1=value1;key2=value2;key3=value3\"." "please see ppdet/utils/profiler.py for detail.") parser.add_argument('--save_proposals', action='store_true', default=False, help='Whether to save the train proposals') parser.add_argument('--proposals_path', type=str, default="sniper/proposals.json", help='Train proposals directory') args = parser.parse_args() return args
def parse_args(): parser = cli.ArgsParser() parser.add_argument( "-c", "--config", type=str, default='configs/yolov3/yolov3_mobilenet_v1_roadsign.yml', help="configuration file to use") parser.add_argument("--eval", action='store_true', default=False, help="Whether to perform evaluation in train") parser.add_argument("-r", "--resume", default=None, help="weights path for resume") parser.add_argument("--slim_config", default=None, type=str, help="Configuration file of slim method.") parser.add_argument("--enable_ce", type=bool, default=False, help="If set True, enable continuous evaluation job." "This flag is only used for " "internal test.") parser.add_argument("--fp16", action='store_true', default=False, help="Enable mixed precision training.") parser.add_argument("--fleet", action='store_true', default=False, help="Use fleet or not") parser.add_argument("--use_vdl", type=bool, default=False, help="whether to record the data to VisualDL.") parser.add_argument('--vdl_log_dir', type=str, default="vdl_log_dir/scalar", help='VisualDL logging directory for scalar.') parser.add_argument('--save_prediction_only', action='store_true', default=False, help='Whether to save the evaluation results only') args = parser.parse_args() return args
def parse_args(): parser = cli.ArgsParser() parser.add_argument( "--weight_type", default='pretrain', type=str, help="Loading Checkpoints only support 'pretrain', 'finetune', 'resume'." ) parser.add_argument( "--fp16", action='store_true', default=False, help="Enable mixed precision training.") parser.add_argument( "--loss_scale", default=8., type=float, help="Mixed precision training loss scale.") parser.add_argument( "--eval", action='store_true', default=False, help="Whether to perform evaluation in train") parser.add_argument( "--output_eval", default=None, type=str, help="Evaluation directory, default is current directory.") parser.add_argument( "--enable_ce", type=bool, default=False, help="If set True, enable continuous evaluation job." "This flag is only used for internal test.") parser.add_argument( "--use_gpu", action='store_true', default=False, help="data parallel") args = parser.parse_args() return args