diff options
| author | Ting-Chun Wang <tcwang0509@berkeley.edu> | 2018-05-30 22:39:01 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-05-30 22:39:01 -0700 |
| commit | a2340c3fff9de44c8ef1fea5b90fced756fbbb18 (patch) | |
| tree | 39f3c05c80a94d721ec6fed0f0da65ecbc3bc603 /options/base_options.py | |
| parent | 1b89cd010dce2e6edaa07d23c8edd8dfe146e0e1 (diff) | |
| parent | 25e205604e7eafa83867a15cfda526461fe58455 (diff) | |
Merge pull request #33 from borisfom/fp16
Added data size and ONNX export options, FP16 inference is working
Diffstat (limited to 'options/base_options.py')
| -rwxr-xr-x | options/base_options.py | 3 |
1 files changed, 2 insertions, 1 deletions
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): |