Skip to content

Commit

Permalink
removed print function
Browse files Browse the repository at this point in the history
  • Loading branch information
BowenD-UCB committed Jul 30, 2024
1 parent 6f7b035 commit b663e80
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions chgnet/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -792,8 +792,6 @@ def forward(
out["s_MAE_size"] = stress_target.shape[0]

# Mag
print(f"{list(prediction)=}")
print(f"{list(targets)=}")
if "m" in self.target_str:
mag_preds, mag_targets = [], []
m_mae_size = 0
Expand Down

0 comments on commit b663e80

Please sign in to comment.