Skip to content

Commit

Permalink
fix dtype for inference model
Browse files Browse the repository at this point in the history
  • Loading branch information
HydrogenSulfate committed Jan 8, 2025
1 parent 4801b2a commit 96d4de3
Show file tree
Hide file tree
Showing 4 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion deepmd/pd/utils/serialization.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def deserialize_to_file(model_file: str, data: dict) -> None:
full_graph=True,
input_spec=[
InputSpec([1, -1, 3], dtype="float64", name="coord"),
InputSpec([1, -1], dtype="int64", name="atype"),
InputSpec([1, -1], dtype="int32", name="atype"),
InputSpec([1, -1, -1], dtype="int32", name="nlist"),
None,
None,
Expand Down
2 changes: 1 addition & 1 deletion source/tests/infer/deeppot_sea.forward_lower.json

Large diffs are not rendered by default.

Binary file modified source/tests/infer/deeppot_sea.forward_lower.pdiparams
Binary file not shown.
Binary file modified source/tests/infer/deeppot_sea.pdiparams
Binary file not shown.

0 comments on commit 96d4de3

Please sign in to comment.