↓ 1 callersMethodcompute_sample_metrics(
self,
batch,
model_traj,
bb_traj,
aa_traj,
clean_aa_traj,
src/byprot/utils/protein/evaluator_dplm2.py:830
↓ 1 callersFunctiondistogram_loss(
logits,
pseudo_beta,
pseudo_beta_mask,
min_bin=2.3125,
max_bin=21.6875,
no_bins=64,
src/byprot/models/structok/modules/loss.py:612
↓ 1 callersFunctionfape_loss(
out: Dict[str, torch.Tensor],
batch: Dict[str, torch.Tensor],
config: ml_collections.ConfigDict,
src/byprot/models/structok/modules/loss.py:355
↓ 1 callersMethodforward(
self,
batch,
encoder_out=None,
tokens=None,
loss_mask=None,
src/byprot/models/dplm/modules/dplm_adapter.py:72
↓ 1 callersFunctionget_initial_dplm2(args, aa_seq, struct_seq, tokenizer, pdb, ori_pdb, device)
src/byprot/utils/scaffold_utils.py:395