removed print message

CederGroupHub · Oct 7, 2023 · 38a5cc4 · 38a5cc4
1 parent 1a558ea
commit 38a5cc4
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/chgnet/model/layers.py b/chgnet/model/layers.py
@@ -250,11 +250,9 @@ def forward(
         new_bond_feas = aggregate(
             bond_update, bond_graph[:, 1], average=False, num_owner=len(bond_feas)
         )
-        print("before mlp", new_bond_feas)
         # New bond features
         if self.use_mlp_out:
             new_bond_feas = self.mlp_out(new_bond_feas)
-        print("after", new_bond_feas)
         if self.resnet:
             new_bond_feas += bond_feas