diff options
| author | Boris Fomitchev <bfomitchev@nvidia.com> | 2018-05-08 00:56:35 -0700 |
|---|---|---|
| committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2018-05-08 00:56:35 -0700 |
| commit | 4ca6b1610f9fa65f8bd7d7c15059bfde18a2f02a (patch) | |
| tree | ec2eeb09cdef6a70ea5612c3e6aa91ed2849414a /options/test_options.py | |
| parent | 736a2dc9afef418820e9c52f4f3b38460360b9f2 (diff) | |
Added data size and ONNX export options, FP16 inference is working
Diffstat (limited to 'options/test_options.py')
| -rwxr-xr-x | options/test_options.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/options/test_options.py b/options/test_options.py index aaeff53..504edf3 100755 --- a/options/test_options.py +++ b/options/test_options.py @@ -12,4 +12,8 @@ class TestOptions(BaseOptions): self.parser.add_argument('--which_epoch', type=str, default='latest', help='which epoch to load? set to latest to use latest cached model') self.parser.add_argument('--how_many', type=int, default=50, help='how many test images to run') self.parser.add_argument('--cluster_path', type=str, default='features_clustered_010.npy', help='the path for clustered results of encoded features') + self.parser.add_argument("--export_onnx", type=str, help="export ONNX model to a given file") + self.parser.add_argument("--engine", type=str, help="run serialized TRT engine") + self.parser.add_argument("--onnx", type=str, help="run ONNX model via TRT") + self.parser.add_argument("-d", "--data_type", default=32, type=int, choices=[8, 16, 32], help="Supported data type i.e. 8, 16, 32 bit") self.isTrain = False |