Skip to content

GitLab

"lib/llm/src/preprocessor/tools/request.rs" did not exist on "4f6f63cd1692d1929bafac54232f608550719aa4"
Commit 15105078 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

added batch_size dimesion to compute_tm

parent a3ea7c65
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/utils/loss.py
View file @ 15105078
...@@ -694,7 +694,7 @@ def compute_tm( ...@@ -694,7 +694,7 @@ def compute_tm(
predicted_tm_term = torch.sum(probs * tm_per_bin, dim=-1) predicted_tm_term = torch.sum(probs * tm_per_bin, dim=-1)
n = residue_weights.shape[-1] n = residue_weights.shape[-1]
pair_mask = residue_weights.new_ones((n, n), dtype=torch.int32) pair_mask = residue_weights.new_ones((1,n, n), dtype=torch.int32)
if interface: if interface:
pair_mask *= (asym_id[..., None] != asym_id[..., None, :]).to(dtype=pair_mask.dtype) pair_mask *= (asym_id[..., None] != asym_id[..., None, :]).to(dtype=pair_mask.dtype)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment