fix: Removed models directory dependencies

Signed-off-by: Anurag Dixit <[email protected]>
pytorch · Dec 16, 2021 · c4413e1 · c4413e1
1 parent 588e1d1
commit c4413e1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/examples/benchmark/py/config/vgg16.yml b/examples/benchmark/py/config/vgg16.yml
@@ -9,7 +9,7 @@ input:
     - 224
   num_inputs: 1
 model: 
-  filename: vgg16_traced.jit.pt
+  filename: models/vgg16_traced.jit.pt
   name: vgg16
 runtime: 
   device: 0

diff --git a/examples/benchmark/py/config/vgg16_trt.yml b/examples/benchmark/py/config/vgg16_trt.yml
@@ -8,7 +8,7 @@ input:
     - 224
   num_inputs: 1
 model: 
-  filename: model.plan
+  filename: models/model.plan
   name: vgg16
 calibration_cache:
   - vgg16.cache

diff --git a/examples/benchmark/py/perf_run.py b/examples/benchmark/py/perf_run.py
@@ -258,7 +258,7 @@ def load_model(params):
     model = None
     is_trt_engine = False
     # Load torch model traced/scripted
-    model_file = os.path.join("models", params.get('model').get('filename'))
+    model_file = params.get('model').get('filename')
 
     if model_file.endswith('.jit.pt'):
         model = torch.jit.load(model_file).cuda()
@@ -278,7 +278,6 @@ def load_model(params):
     parser = ConfigParser(args.config)
     # Load YAML params
     params = parser.read_config()
-
     print("Loading model: ", params.get('model').get('filename'))
 
     model = None