Changed export from edge index to full data object for better reprodu…

…ction of errors
elliottower · Apr 16, 2021 · b6e8139 · b6e8139
1 parent d86c557
commit b6e8139
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ocpmodels/models/graphtransformer.py b/ocpmodels/models/graphtransformer.py
@@ -541,7 +541,7 @@ def convert_input(args, data):
     torch.save(a2a2, path + "/a2a2.pt")
     torch.save(a2b2, path + "/a2b2.pt")
     torch.save(b2a2, path + "/b2a2.pt")
-    torch.save(data.edge_index, path + "/edge_index.pt")
+    torch.save(data, path + "/data.pt")
 
     # Set them equal temporarily
     a2b = a2b2