Skip to content

Commit

Permalink
minor fix network choice
Browse files Browse the repository at this point in the history
  • Loading branch information
qiaohaijun committed Jan 12, 2016
1 parent 08466f8 commit 120dd4b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion example/image-classification/train_imagenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand Down

0 comments on commit 120dd4b

Please sign in to comment.