From 4ca6b1610f9fa65f8bd7d7c15059bfde18a2f02a Mon Sep 17 00:00:00 2001 From: Boris Fomitchev Date: Tue, 8 May 2018 00:56:35 -0700 Subject: Added data size and ONNX export options, FP16 inference is working --- options/base_options.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'options/base_options.py') diff --git a/options/base_options.py b/options/base_options.py index de831fe..561a890 100755 --- a/options/base_options.py +++ b/options/base_options.py @@ -56,7 +56,8 @@ class BaseOptions(): self.parser.add_argument('--n_downsample_E', type=int, default=4, help='# of downsampling layers in encoder') self.parser.add_argument('--nef', type=int, default=16, help='# of encoder filters in the first conv layer') self.parser.add_argument('--n_clusters', type=int, default=10, help='number of clusters for features') - + self.parser.add_argument('--verbose', action='store_true', default = False, help='toggles verbose') + self.initialized = True def parse(self, save=True): -- cgit v1.2.3-70-g09d2