Skip to content

GitLab

Commit 0caffd57 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

fixed the error where the iterface backbone loss falsely used intra_chain backbone weights

parent bc35ef17
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/utils/loss.py
View file @ 0caffd57
...@@ -310,10 +310,10 @@ def fape_loss( ...@@ -310,10 +310,10 @@ def fape_loss(
interface_bb_loss = backbone_loss( interface_bb_loss = backbone_loss(
traj=traj, traj=traj,
pair_mask=1. - intra_chain_mask, pair_mask=1. - intra_chain_mask,
**{**batch, **config.intra_chain_backbone}, **{**batch, **config.interface},
) )
weighted_bb_loss = (intra_chain_bb_loss * config.intra_chain_backbone.weight weighted_bb_loss = (intra_chain_bb_loss * config.intra_chain_backbone.weight
+ interface_bb_loss * config.intra_chain_backbone.weight) + interface_bb_loss * config.interface.weight)
else: else:
bb_loss = backbone_loss( bb_loss = backbone_loss(
traj=traj, traj=traj,
......
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