Skip to content

GitLab

Commit 5348936a authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

now update get_optimal_transform

parent 9f24ebf0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
openfold/utils/loss.py
View file @ 5348936a
...@@ -2069,7 +2069,7 @@ class AlphaFoldMultimerLoss(AlphaFoldLoss): ...@@ -2069,7 +2069,7 @@ class AlphaFoldMultimerLoss(AlphaFoldLoss):
# anchor_pred_mask = anchor_pred_mask.to('cuda') # anchor_pred_mask = anchor_pred_mask.to('cuda')
input_mask = (anchor_true_mask * anchor_pred_mask).bool() input_mask = (anchor_true_mask * anchor_pred_mask).bool()
r, x = get_optimal_transform( r, x = get_optimal_transform(
anchor_true_pos, anchor_true_pos[0],
anchor_pred_pos,mask=input_mask anchor_pred_pos,mask=input_mask
) )
del input_mask # just to save memory del input_mask # just to save memory
......
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