Update repflows.py

iProzd · Jan 22, 2025 · 7c9a5c2 · 7c9a5c2
1 parent af999b1
commit 7c9a5c2
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/deepmd/pt/model/descriptor/repflows.py b/deepmd/pt/model/descriptor/repflows.py
@@ -540,8 +540,6 @@ def forward(
         for idx, ll in enumerate(self.layers):
             # node_ebd:     nb x nloc x n_dim
             # node_ebd_ext: nb x nall x n_dim
-            # print(idx, "node_ebd.std(): ", node_ebd.std().item(), "edge_ebd.std(): ", edge_ebd.std().item(),
-            #       "angle_ebd.std(): ", angle_ebd.std().item())
             if comm_dict is None:
                 assert mapping is not None
                 assert mapping3 is not None