Fix issue in TM computation

gahdritz · web-flow · commit 84659c93ba6f · 2023-02-15T22:27:05.000-05:00
diff --git a/openfold/utils/loss.py b/openfold/utils/loss.py
@@ -640,9 +640,7 @@ def compute_tm(
     )
 
     bin_centers = _calculate_bin_centers(boundaries)
-    torch.sum(residue_weights)
-    n = logits.shape[-2]
-    clipped_n = max(n, 19)
+    clipped_n = max(torch.sum(residue_weights), 19)
 
     d0 = 1.24 * (clipped_n - 15) ** (1.0 / 3) - 1.8
 

Original file line number	Diff line number	Diff line change
`@@ -640,9 +640,7 @@ def compute_tm(`
`640`	`640`	`)`
`641`	`641`
`642`	`642`	`bin_centers = _calculate_bin_centers(boundaries)`
`643`		`- torch.sum(residue_weights)`
`644`		`- n = logits.shape[-2]`
`645`		`- clipped_n = max(n, 19)`
	`643`	`+ clipped_n = max(torch.sum(residue_weights), 19)`
`646`	`644`
`647`	`645`	`d0 = 1.24 * (clipped_n - 15) ** (1.0 / 3) - 1.8`
`648`	`646`