[minor] fixed typo in save_load

mit-han-lab · Nov 24, 2023 · 60b055f · 60b055f
1 parent 1637680
commit 60b055f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/save_load_example/save_load.py b/examples/save_load_example/save_load.py
@@ -143,7 +143,7 @@ def save_load3():
     # print(model.q_layer.rx0._parameters)
 
     traced_cell = torch.jit.trace(model, (x))
-    torch.jit.save(traced_cell, "model_trace.pth")
+    torch.jit.save(traced_cell, "model_trace.pt")
 
     loaded_trace = torch.jit.load("model_trace.pt")
     y2 = loaded_trace(x)