模型训练----parser.add_argument添加配置参数

parser = argparse.ArgumentParser() parser.add_argument("--weights_path", type=str, default= 'weights_lr_1e-7') parser.add_argument("--num_classes", type=int, default=8) parser.add_argument("--epochs", type=int, default=1000) parser.add_argument("--batch-size", type=int, default=8) parser.add_argument("--lr", type=float, default=0.0000001) parser.add_argument("--lrf", type=float, default=0.01) 小白，解释一下这段是什么意思

import argparse def args_parser(): parser = argparse.ArgumentParser() ### new parser.add_argument('--exp', type=str, default='VEIN/UNET_FL', help='experiment_name') parser.add_argument('--labeled_num', type=int, default=1000, help='labeled data') parser.add_argument('--model', type=str, default='unet_fl', help='model_name') parser.add_argument('--num_classes', type=int, default=2, help='output channel of network') parser.add_argument('--gpu', type=str, default='0,1', help='GPU to use') parser.add_argument('--deterministic', type=int, default=1, help='whether use deterministic training') parser.add_argument('--seed', type=int, default=1337, help='random seed') parser.add_argument('--root_path', type=str, default='/data3/vein/vein-train-1000-480-640', help='dataset root dir') parser.add_argument('--num_users', type=int, default=10, help='local epoch') parser.add_argument('--batch_size', type=int, default=4, help='batch_size per gpu') parser.add_argument('--base_lr', type=float, default=1e-4, help='segmentation network learning rate') parser.add_argument('--max_iterations', type=int, default=1000, help='maximum epoch number to train') parser.add_argument('--rounds', type=int, default=50, help='local epoch') parser.add_argument('--local_ep', type=int, default=1, help='local epoch') parser.add_argument('--dataset_seg', type=str, default='vein', help='dataset') parser.add_argument('--in_chan', type=int, default=1, help='input channel') parser.add_argument('--base_chan', type=int, default=32, help='base channel') parser.add_argument('--out_chan', type=int, default=2, help='out channel') parser.add_argument('--reduce_size', type=list, default=[9,12], help='reduce size') #parser.add_argument('--local_train', type=int, default=1000, help='local train time') parser.add_argument('--root_path_unlabeled', type=str, default='/data3/vein/vein-train-unlabel-2000-480-640', help='dataset root dir of unlabeled data') parser.add_argument('--each_length', type=int, default=225, help='data number each length') parser.add_argument('--dist_scale', type=int, default=0.01, help='scale factor when computing model distance') parser.add_argument("--max_grad_norm", dest="max_grad_norm", type=float, default=5, help="max gradient norm allowed (used for gradient clipping)") parser.add_argument('--num-warmup-epochs', '--num-warm-up-epochs', dest="num_warmup_epochs", default=0, type=int, help='number of warm-up epochs for unsupervised loss ramp-up during training' 'set to 0 to disable ramp-up') parser.add_argument('--lambda_u', type=float, default=100, help='start_epoch') parser.add_argument('--ema_decay', type=float, default=0.999, help='ema_decay') parser.add_argument('--consistency', type=float, default=0.1, help='consistency') parser.add_argument('--consistency_rampup', type=float, default=200.0, help='consistency_rampup') parser.add_argument('--labeled_bs', type=int, default=4, help='labeled_batch_size per gpu') parser.add_argument('--ict_alpha', type=int, default=0.2, help='ict_alpha') args = parser.parse_args() return args

分析代码：if __name__ == "__main__": mp.set_start_method("spawn") torch.backends.cuda.matmul.allow_tf32 = True torch.set_grad_enabled(False) device = "cuda:0" save_frames = False datetime_now = str(datetime.datetime.now()).replace(" ", "_") parser = argparse.ArgumentParser() parser.add_argument("--dataset", default="datasets/tum/rgbd_dataset_freiburg1_desk") parser.add_argument("--config", default="config/base.yaml") parser.add_argument("--save-as", default="default") parser.add_argument("--no-viz", action="store_true") parser.add_argument("--calib", default="")

分析并注释每一行 """参数解析""" import os import argparse import ast def train_parse_args(): train_parser = argparse.ArgumentParser(description='Image classification trian') train_parser.add_argument('--platform', type=str, default="CPU", choices=("CPU", "GPU", "Ascend"), \ help='run platform, only support CPU, GPU and Ascend') train_parser.add_argument('--dataset_path', type=str, default="dataset\PetImages", help='Dataset path') train_parser.add_argument('--pretrain_ckpt', type=str, default="mobilenetV2.ckpt", help='Pretrained checkpoint path \ for fine tune or incremental learning') train_parser.add_argument('--freeze_layer', type=str, default="backbone", choices=["", "none", "backbone"], \ help="freeze the weights of network from start to which layers") train_parser.add_argument('--run_distribute', type=ast.literal_eval, default=False, help='Run distribute') train_parser.add_argument('--filter_head', type=ast.literal_eval, default=False,\ help='Filter head weight parameters when load checkpoint, default is False.') train_args = train_parser.parse_args() train_args.is_training = True if train_args.platform == "CPU": train_args.run_distribute = False train_args.dataset_path = os.path.abspath(train_args.dataset_path) return train_args

parser = argparse.ArgumentParser() parser.add_argument("--weights", type=str, default=ROOT / "yolov5s.pt", help="initial weights path") parser.add_argument("--cfg", type=str, default="", help="model.yaml path") parser.add_argument("--data", type=str, default=ROOT / "D:/project/py/YOLOpythonProject1/yolov5/data/images/data.yaml", help="D:/project/py/YOLOpythonProject1/yolov5/data/images") parser.add_argument("--hyp", type=str, default=ROOT / "data/hyps/hyp.scratch-low.yaml", help="hyperparameters path") parser.add_argument("--epochs", type=int, default=100, help="total training epochs") parser.add_argument("--batch-size", type=int, default=16, help="total batch size for all GPUs, -1 for autobatch") parser.add_argument("--imgsz", "--img", "--img-size", type=int, default=640, help="train, val image size (pixels)") parser.add_argument("--rect", action="store_true", help="rectangular training") parser.add_argument("--resume", nargs="?", const=True, default=False, help="resume most recent training") parser.add_argument("--nosave", action="store_true", help="only save final checkpoint") parser.add_argument("--noval", action="store_true", help="only validate final epoch") parser.add_argument("--noautoanchor", action="store_true", help="disable AutoAnchor") parser.add_argument("--noplots", action="store_true", help="save no plot files") parser.add_argument("--evolve", type=int, nargs="?", const=300, help="evolve hyperparameters for x generations") parser.add_argument( "--evolve_population", type=str, default=ROOT / "data/hyps", help="location for loading population" ) 怎么改