Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
njzjz authored Mar 6, 2024
1 parent 96683c7 commit 9d7fea7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepmd/pt/loss/tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def forward(self, model_pred, label, natoms, learning_rate=0.0, mae=False):
Other losses for display.
"""
del learning_rate, mae
loss = torch.tensor.zeros(
loss = torch.zeros(
1, dtype=env.GLOBAL_PT_FLOAT_PRECISION, device=env.DEVICE
)[0]
more_loss = {}
Expand Down

0 comments on commit 9d7fea7

Please sign in to comment.