From 91e22138f3bc8df0e66af7b0f34dd4a56baad7cf Mon Sep 17 00:00:00 2001 From: Raphael Delhome Date: Tue, 27 Mar 2018 17:06:20 +0200 Subject: [PATCH] Update docstrings regarding chosen network architecture --- sources/train.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/sources/train.py b/sources/train.py index b4475371..c25b6edd 100644 --- a/sources/train.py +++ b/sources/train.py @@ -61,10 +61,13 @@ help=("Model name that will be used for results, " "checkout and graph storage on file system")) parser.add_argument('-N', '--network', default='small', - help=("Neural network size, either 'small' or 'medium'" + help=("Neural network size, either 'small', 'medium'," + " 'vgg' or 'inception' " "('small' refers to 3 conv/pool blocks and 1 " - "fully-connected layer, and 'medium' refers to 6" - "conv/pool blocks and 2 fully-connected layers)")) # TO UPDATE + "fully-connected layer; 'medium' refers to 6" + "conv/pool blocks and 2 fully-connected layers;" + " 'vgg' and 'inception' refer to " + "state-of-the-art networks)")) parser.add_argument('-r', '--learning-rate', required=False, nargs="+", default=[0.01, 1000, 0.95], type=float, help=("List of learning rate components (starting LR, " @@ -127,10 +130,10 @@ "each batch (convex weights with min at 50%)...")) sys.exit(1) - if not args.network in ["small", "medium", "vgg", "inception", "inception-v2", "inception-v3"]: + if not args.network in ["small", "medium", "vgg", "inception"]: utils.logger.error(("Unsupported network size description" - "Please use this parameter with 'small' or " - "'medium' values")) + "Please use this parameter with 'small', " + "'medium', 'vgg' or 'inception' values")) sys.exit(1) if not args.model in ["feature_detection", "semantic_segmentation"]: