Merge pull request pfnet-research#4 from mwata/input_names

Add input names
shinh · Mar 31, 2023 · 3fa66ac · 3fa66ac
2 parents c59b903 + 9713fa7
commit 3fa66ac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/export_static_onnx.py b/export_static_onnx.py
@@ -134,4 +134,5 @@ def prepare_data(args):
     print("energy = ", float(exporter.forward(**args)), "eV")
 
     print("out_dir = ", out_dir, file=sys.stderr)
-    ppe_onnx.export_testcase(exporter, tuple(args.values()), out_dir, verbose=True)
+    ppe_onnx.export_testcase(exporter, tuple(args.values()), out_dir, verbose=True,
+                             input_names=["Z","pos","shift_vecs","cell_volume","triu"])