From 11b6899afa7ec04a3678920fdce10ff0c14cde67 Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Wed, 28 Aug 2019 01:41:42 +0900 Subject: [PATCH] Modified the default value for arch There are no 'vovnet69v2' in vovnet. So I changed this to 'vovnet57'. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 630167b..fad59de 100644 --- a/main.py +++ b/main.py @@ -26,7 +26,7 @@ parser = argparse.ArgumentParser(description='PyTorch ImageNet Training') parser.add_argument('data', metavar='DIR', help='path to dataset') -parser.add_argument('--arch', '-a', metavar='ARCH', default='vovnet69v2', +parser.add_argument('--arch', '-a', metavar='ARCH', default='vovnet57', #choices=model_names, help='model architecture: ' + ' | '.join(model_names) +