Skip to content

Commit 4f2e001

Browse files
authored
Merge pull request #93 from amcadmus/devel
fix bug in dp test, inconsistent array shape
2 parents 2fc61fa + 48b60f9 commit 4f2e001

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)