Skip to content

Commit 0de1276

Browse files
authored
Merge pull request #133 from amcadmus/devel
fix bug of model attributes for se_ar
2 parents c790560 + e1f5a36 commit 0de1276

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

source/train/DescrptSeAR.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def build (self,
7272
reuse = None):
7373
# dout
7474
self.dout_a = self.descrpt_a.build(coord_, atype_, natoms, box, mesh, davg[0], dstd[0], suffix=suffix+'_a', reuse=reuse)
75-
self.dout_r = self.descrpt_r.build(coord_, atype_, natoms, box, mesh, davg[1], dstd[1], suffix=suffix+'_r', reuse=reuse)
75+
self.dout_r = self.descrpt_r.build(coord_, atype_, natoms, box, mesh, davg[1], dstd[1], suffix=suffix, reuse=reuse)
7676
self.dout_a = tf.reshape(self.dout_a, [-1, self.descrpt_a.get_dim_out()])
7777
self.dout_r = tf.reshape(self.dout_r, [-1, self.descrpt_r.get_dim_out()])
7878
self.dout = tf.concat([self.dout_a, self.dout_r], axis = 1)

0 commit comments

Comments
 (0)