We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f8f0ab7 commit 21a9832Copy full SHA for 21a9832
tests/py/dynamo/models/test_export_serde.py
@@ -43,9 +43,9 @@ def forward(self, x):
43
44
exp_program = torchtrt.dynamo.trace(model, **compile_spec)
45
trt_module = torchtrt.dynamo.compile(exp_program, **compile_spec)
46
- torchtrt.save(trt_module, "/tmp/trt.ep", inputs=[input])
+ torchtrt.save(trt_module, "./trt.ep", inputs=[input])
47
# TODO: Enable this serialization issues are fixed
48
- # deser_trt_module = torchtrt.load("/tmp/trt.ep").module()
+ # deser_trt_module = torchtrt.load("./trt.ep").module()
49
# Check Pyt and TRT exported program outputs
50
cos_sim = cosine_similarity(model(input), trt_module(input)[0])
51
assertions.assertTrue(
0 commit comments