Skip to content

Commit c4413e1

Browse files
author
Anurag Dixit
committed
fix: Removed models directory dependencies
Signed-off-by: Anurag Dixit <[email protected]>
1 parent 588e1d1 commit c4413e1

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

Diff for: examples/benchmark/py/config/vgg16.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ input:
99
- 224
1010
num_inputs: 1
1111
model:
12-
filename: vgg16_traced.jit.pt
12+
filename: models/vgg16_traced.jit.pt
1313
name: vgg16
1414
runtime:
1515
device: 0

Diff for: examples/benchmark/py/config/vgg16_trt.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ input:
88
- 224
99
num_inputs: 1
1010
model:
11-
filename: model.plan
11+
filename: models/model.plan
1212
name: vgg16
1313
calibration_cache:
1414
- vgg16.cache

Diff for: examples/benchmark/py/perf_run.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,7 @@ def load_model(params):
258258
model = None
259259
is_trt_engine = False
260260
# Load torch model traced/scripted
261-
model_file = os.path.join("models", params.get('model').get('filename'))
261+
model_file = params.get('model').get('filename')
262262

263263
if model_file.endswith('.jit.pt'):
264264
model = torch.jit.load(model_file).cuda()
@@ -278,7 +278,6 @@ def load_model(params):
278278
parser = ConfigParser(args.config)
279279
# Load YAML params
280280
params = parser.read_config()
281-
282281
print("Loading model: ", params.get('model').get('filename'))
283282

284283
model = None

0 commit comments

Comments
 (0)