start working on loss part

dingquanyu · Jun 27, 2023 · eeb035c · eeb035c
1 parent 24e470b
commit eeb035c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/openfold/utils/loss.py b/openfold/utils/loss.py
@@ -2083,6 +2083,8 @@ def forward(self,out,batch,_return_breakdown=False):
  # then permutate ground truth chains before calculating the loss
  permutated_labels = self.multi_chain_perm_align(out,features,labels)
  logger.info("finished multi-chain permutation")
+ # features.update(permutated_labels)
+ # self.loss(out,features)
  return permutated_labels
  ## TODO next need to check how the ground truth label is used
  # in loss calculation.