We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 2fc61fa + 48b60f9 commit 4f2e001Copy full SHA for 4f2e001
source/train/test.py
@@ -49,7 +49,7 @@ def test_ener (args) :
49
ae = ae.reshape([numb_test,-1])
50
av = av.reshape([numb_test,-1])
51
52
- l2e = (l2err (energy - test_data["energy"][:numb_test]))
+ l2e = (l2err (energy - test_data["energy"][:numb_test].reshape([-1,1])))
53
l2f = (l2err (force - test_data["force"] [:numb_test]))
54
l2v = (l2err (virial - test_data["virial"][:numb_test]))
55
l2ea= l2e/natoms
0 commit comments