Skip to content

Commit e20b2d4

Browse files
committed
remove GLOBAL_TF_FLOAT_PRECISION
1 parent 9dfb1fa commit e20b2d4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

deepmd/tf/nvnmd/entrypoints/mapt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -614,13 +614,13 @@ def build_t2g(self):
614614
padding_ntypes = type_embedding.shape[0]
615615
type_embedding_nei = tf.tile(
616616
tf.reshape(
617-
type_embedding, [1, padding_ntypes, -1], dtype=GLOBAL_TF_FLOAT_PRECISION
617+
type_embedding, [1, padding_ntypes, -1]
618618
),
619619
[padding_ntypes, 1, 1],
620620
) # (ntypes) * ntypes * Y
621621
type_embedding_center = tf.tile(
622622
tf.reshape(
623-
type_embedding, [padding_ntypes, 1, -1], dtype=GLOBAL_TF_FLOAT_PRECISION
623+
type_embedding, [padding_ntypes, 1, -1]
624624
),
625625
[1, padding_ntypes, 1],
626626
) # ntypes * (ntypes) * Y

0 commit comments

Comments
 (0)