|
|
@@ -69,7 +69,7 @@ parser.add_argument('--prune_save', default='prune_best.pt', type=str, help='|
|
|
|
|
|
|
# 模型处理的部分了
|
|
|
parser.add_argument('--timm', default=False, type=bool, help='|是否使用timm库创建模型|')
|
|
|
-parser.add_argument('--model', default='mobilenetv2', type=str, help='|自定义模型选择，timm为True时为timm库中模型|')
|
|
|
+parser.add_argument('--model', default='mobilenetv2', type=str, help='|自定义模型选择，LeNet,Alexnet,VGG19,GoogleNet,resnet中选择|')
|
|
|
parser.add_argument('--model_type', default='s', type=str, help='|自定义模型型号|')
|
|
|
parser.add_argument('--save_path', default=f'${pwd}/checkpoints/mobilenetv2/best.pt', type=str, help='|保存最佳模型，除此之外每轮还会保存last.pt|')
|
|
|
parser.add_argument('--save_path_last', default=f'${pwd}/checkpoints/mobilenetv2/last.pt', type=str, help='|保存最佳模型，除此之外每轮还会保存last.pt|')
|
