diff options
Diffstat (limited to 'candle-core/tests/pth.py')
| -rw-r--r-- | candle-core/tests/pth.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-core/tests/pth.py b/candle-core/tests/pth.py index 97724712..cab94f2c 100644 --- a/candle-core/tests/pth.py +++ b/candle-core/tests/pth.py @@ -6,3 +6,5 @@ a= torch.tensor([[1,2,3,4], [5,6,7,8]]) o = OrderedDict() o["test"] = a torch.save(o, "test.pt") + +torch.save({"model_state_dict": o}, "test_with_key.pt") |