@@ -24,7 +24,6 @@ def test_resnet18(ir):
24
24
"device" : torchtrt .Device ("cuda:0" ),
25
25
"enabled_precisions" : {torch .float },
26
26
"ir" : ir ,
27
- "max_num_trt_engines" : 200 ,
28
27
}
29
28
30
29
trt_mod = torchtrt .compile (model , ** compile_spec )
@@ -55,7 +54,6 @@ def test_mobilenet_v2(ir):
55
54
"device" : torchtrt .Device ("cuda:0" ),
56
55
"enabled_precisions" : {torch .float },
57
56
"ir" : ir ,
58
- "max_num_trt_engines" : 200 ,
59
57
}
60
58
61
59
trt_mod = torchtrt .compile (model , ** compile_spec )
@@ -86,7 +84,6 @@ def test_efficientnet_b0(ir):
86
84
"device" : torchtrt .Device ("cuda:0" ),
87
85
"enabled_precisions" : {torch .float },
88
86
"ir" : ir ,
89
- "max_num_trt_engines" : 200 ,
90
87
}
91
88
92
89
trt_mod = torchtrt .compile (model , ** compile_spec )
@@ -126,7 +123,6 @@ def test_bert_base_uncased(ir):
126
123
"enabled_precisions" : {torch .float },
127
124
"truncate_long_and_double" : True ,
128
125
"ir" : ir ,
129
- "max_num_trt_engines" : 200 ,
130
126
}
131
127
trt_mod = torchtrt .compile (model , ** compile_spec )
132
128
0 commit comments