From 120dd4b78154a80d64d5e019969f96476176a6f5 Mon Sep 17 00:00:00 2001 From: qiao hai-jun Date: Tue, 12 Jan 2016 22:35:46 +0800 Subject: [PATCH] minor fix network choice --- example/image-classification/train_imagenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/image-classification/train_imagenet.py b/example/image-classification/train_imagenet.py index 54bf605e5d28..dfd573a4e17e 100644 --- a/example/image-classification/train_imagenet.py +++ b/example/image-classification/train_imagenet.py @@ -7,7 +7,7 @@ # don't use -n and -s, which are resevered for the distributed training parser = argparse.ArgumentParser(description='train an image classifer on imagenet') parser.add_argument('--network', type=str, default='inception-bn', - choices = ['alexnet', 'vgg', 'googlenet', 'inception-bn', 'inception-bn-full.py'], + choices = ['alexnet', 'vgg', 'googlenet', 'inception-bn', 'inception-bn-full'], help = 'the cnn to use') parser.add_argument('--data-dir', type=str, required=True, help='the input data directory')