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) +