File tree Expand file tree Collapse file tree 3 files changed +3
-4
lines changed Expand file tree Collapse file tree 3 files changed +3
-4
lines changed Original file line number Diff line number Diff line change 9
9
- 224
10
10
num_inputs : 1
11
11
model :
12
- filename : vgg16_traced.jit.pt
12
+ filename : models/ vgg16_traced.jit.pt
13
13
name : vgg16
14
14
runtime :
15
15
device : 0
Original file line number Diff line number Diff line change 8
8
- 224
9
9
num_inputs : 1
10
10
model :
11
- filename : model.plan
11
+ filename : models/ model.plan
12
12
name : vgg16
13
13
calibration_cache :
14
14
- vgg16.cache
Original file line number Diff line number Diff line change @@ -258,7 +258,7 @@ def load_model(params):
258
258
model = None
259
259
is_trt_engine = False
260
260
# 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' )
262
262
263
263
if model_file .endswith ('.jit.pt' ):
264
264
model = torch .jit .load (model_file ).cuda ()
@@ -278,7 +278,6 @@ def load_model(params):
278
278
parser = ConfigParser (args .config )
279
279
# Load YAML params
280
280
params = parser .read_config ()
281
-
282
281
print ("Loading model: " , params .get ('model' ).get ('filename' ))
283
282
284
283
model = None
You can’t perform that action at this time.
0 commit comments