Unverified Commit a885bb73 authored by Yuxin Wu's avatar Yuxin Wu Committed by GitHub
Browse files

Update test_checkpoint.py

parent 263ddafe
......@@ -42,7 +42,7 @@ class TestCheckpointer(unittest.TestCase):
# different tensor references
self.assertFalse(id(loaded) == id(stored))
# same content
self.assertTrue(loaded.equal(stored))
self.assertTrue(loaded.to(stored).equal(stored))
if __name__ == "__main__":
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment