Skip to content

Commit 7268d48

Browse files
authored
loss_pf do not needs to be divided by atom number
1 parent e12373e commit 7268d48

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

source/train/Loss.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def print_on_training(self,
169169
if self.has_v :
170170
print_str += prop_fmt % (np.sqrt(error_v_test) / natoms[0], np.sqrt(error_v_train) / natoms[0])
171171
if self.has_pf:
172-
print_str += prop_fmt % (np.sqrt(error_pf_test) / natoms[0], np.sqrt(error_pf_train) / natoms[0])
172+
print_str += prop_fmt % (np.sqrt(error_pf_test), np.sqrt(error_pf_train))
173173

174174
return print_str
175175

0 commit comments

Comments
 (0)