Skip to content

Commit 48b60f9

Browse files
author
Han Wang
committed
fix bug in dp test, inconsistent array shape
1 parent d9af7d3 commit 48b60f9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

source/train/test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def test_ener (args) :
4949
ae = ae.reshape([numb_test,-1])
5050
av = av.reshape([numb_test,-1])
5151

52-
l2e = (l2err (energy - test_data["energy"][:numb_test]))
52+
l2e = (l2err (energy - test_data["energy"][:numb_test].reshape([-1,1])))
5353
l2f = (l2err (force - test_data["force"] [:numb_test]))
5454
l2v = (l2err (virial - test_data["virial"][:numb_test]))
5555
l2ea= l2e/natoms

0 commit comments

Comments
 (0)