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
......@@ -310,10 +310,10 @@ def fape_loss(
interface_bb_loss = backbone_loss(
traj=traj,
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
+ interface_bb_loss * config.intra_chain_backbone.weight)
+ interface_bb_loss * config.interface.weight)
else:
bb_loss = backbone_loss(
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