From 649beefdc00d05870952880a2c91985a3e7606d4 Mon Sep 17 00:00:00 2001 From: root Date: Fri, 21 Apr 2023 02:33:53 +0000 Subject: [PATCH] updata --model parameter --- cv/classification/resnet50/pytorch/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv/classification/resnet50/pytorch/train.py b/cv/classification/resnet50/pytorch/train.py index eb61e7d9b..a19d1f753 100644 --- a/cv/classification/resnet50/pytorch/train.py +++ b/cv/classification/resnet50/pytorch/train.py @@ -342,7 +342,7 @@ def get_args_parser(add_help=True): parser = argparse.ArgumentParser(description='PyTorch Classification Training', add_help=add_help) parser.add_argument('--data-path', default='/datasets01/imagenet_full_size/061417/', help='dataset') - parser.add_argument('--model', default='resnet18', help='model') + parser.add_argument('--model', default='resnet50', help='model') parser.add_argument('--device', default='cuda', help='device') parser.add_argument('-b', '--batch-size', default=32, type=int) parser.add_argument('--epochs', default=90, type=int, metavar='N', -- Gitee