API for torch.load() #2252
iprovalo
started this conversation in
Development
Replies: 2 comments 2 replies
-
We only support load numpy format, you can convert it to numpy first and save it in either .npy or .npz format. |
Beta Was this translation helpful? Give feedback.
1 reply
-
You can use:
|
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I need to be able to read a serialized shared non-layer specific tensor (weights) which is saved like so:
torch.save(model.shared.weight, 'weights.bin')
In Python it can be read:
torch.load('weights.bin')
Is there a way to load this tensor in DJL?
Thank you!
Beta Was this translation helpful? Give feedback.
All reactions