Skip to content

Commit

Permalink
ckean OutputVariableDef
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed Feb 2, 2024
1 parent e53b476 commit 5b16abf
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 10 deletions.
5 changes: 0 additions & 5 deletions deepmd/infer/deep_pot.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,11 +63,6 @@ def output_def(self) -> ModelOutputDef:
differentiable=True,
atomic=True,
),
OutputVariableDef(
# ugly...
"energy_derv_c_redu",
shape=[1, 3, 3],
),
]
)
)
Expand Down
5 changes: 0 additions & 5 deletions deepmd/infer/deep_tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -230,11 +230,6 @@ def output_def(self) -> ModelOutputDef:
differentiable=True,
atomic=True,
),
OutputVariableDef(
# ugly...
f"{self.output_tensor_name}_derv_c_redu",
shape=[-1, 3, 3],
),
]
)
)

0 comments on commit 5b16abf

Please sign in to comment.