We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e0b517 commit 5db7a56Copy full SHA for 5db7a56
tests/data/meta_tensor/test_meta_tensor.py
@@ -245,7 +245,7 @@ def test_pickling(self):
245
with tempfile.TemporaryDirectory() as tmp_dir:
246
fname = os.path.join(tmp_dir, "im.pt")
247
torch.save(m, fname)
248
- m2 = torch.load(fname, weights_only=True)
+ m2 = torch.load(fname, weights_only=False)
249
self.check(m2, m, ids=False)
250
251
@skip_if_no_cuda
0 commit comments