metric bug

Cloud303 · Aug 14, 2024 · d87a740 · d87a740
1 parent 72442b3
commit d87a740
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models/model_wrapper.py b/src/models/model_wrapper.py
@@ -144,7 +144,7 @@ def _compute_validation_metrics(
             batch_size, n_tokens = batch["residue_index"].shape
             metrics = {}
 
-            gt_coords = outputs["all_atom_positions"]  # (bs, n_atoms, 3) gt_atoms after augmentation
+            gt_coords = batch["all_atom_positions"]  # (bs, n_atoms, 3) gt_atoms after augmentation
             pred_coords = outputs["sampled_positions"].squeeze(-3)  # remove S dimension (bs, 1, n_atoms, 3)
             all_atom_mask = batch["atom_mask"]  # (bs, n_atoms)