From 60b055f0ff96c38488f97cff66a93e7cf9041afc Mon Sep 17 00:00:00 2001 From: GenericP3rson Date: Fri, 24 Nov 2023 16:47:19 -0500 Subject: [PATCH] [minor] fixed typo in save_load --- examples/save_load_example/save_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/save_load_example/save_load.py b/examples/save_load_example/save_load.py index 1022c3aa..c5a6c57a 100644 --- 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)