Add testing for NJT serialization #126453
Labels
module: nestedtensor
NestedTensor tag see issue #25032
module: serialization
Issues related to serialization (e.g., via pickle, or otherwise) of PyTorch objects
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃悰 Describe the bug
From discussion with @jbschlosser there currently might not be testing for NJT serialization/deserialization (but there seems to be testing for NST), filing issue to add testing for NJT so we can test for compatibility with weights_only etc.
Versions
main
cc @mruberry @msaroufim @cpuhrsch @jbschlosser @bhosmer @drisspg @soulitzer
The text was updated successfully, but these errors were encountered: