Skip to content

Commit 8448dfa

Browse files
committed
chore: minor fixes
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 3f1a655 commit 8448dfa

File tree

2 files changed

+2
-8
lines changed

2 files changed

+2
-8
lines changed

tests/cpp/test_collections.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -192,9 +192,9 @@ TEST(CppAPITests, TestCollectionTupleInputOutput) {
192192
auto trt_out = trt_mod.forward(complex_inputs);
193193

194194
ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(
195-
out.toTuple()->elements()[0].toTensor(), trt_out.toTuple()->elements()[0].toTensor(), 1e-5));
195+
out.toTuple()->elements()[0].toTensor(), trt_out.toTuple()->elements()[0].toTensor(), 1e-4));
196196
ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(
197-
out.toTuple()->elements()[1].toTensor(), trt_out.toTuple()->elements()[1].toTensor(), 1e-5));
197+
out.toTuple()->elements()[1].toTensor(), trt_out.toTuple()->elements()[1].toTensor(), 1e-4));
198198
}
199199

200200
TEST(CppAPITests, TestCollectionListInputOutput) {

tests/modules/hub.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -46,12 +46,6 @@
4646
"model": torch.hub.load("pytorch/vision:v0.9.0", "resnet50", pretrained=True),
4747
"path": "both",
4848
},
49-
"ssd": {
50-
"model": torch.hub.load(
51-
"NVIDIA/DeepLearningExamples:torchhub", "nvidia_ssd", model_math="fp32"
52-
),
53-
"path": "trace",
54-
},
5549
"efficientnet_b0": {
5650
"model": timm.create_model("efficientnet_b0", pretrained=True),
5751
"path": "script",

0 commit comments

Comments
 (0)