↓ 2 callersMethod__init__(self, input_nf, output_nf, hidden_nf, n_channel, edges_in_d=0, act_fn=nn.SiLU(),
residual=Tr
models/MCAttGNN/mc_egnn.py:23
↓ 2 callersMethod__init__(self, embed_size, hidden_size, n_channel, n_edge_feats=0,
n_layers=3, cdr_type='3', alpha=0.
models/MCAttGNN/mc_att_model.py:142
↓ 2 callersFunction_alpha_from_logits Args: logits: Logit matrices, (N, L_i, L_j, num_heads). mask: Masks, (N, L). Returns: alpha: Attention weights.
evaluation/ddg/models/attention.py:9
↓ 2 callersFunctionconstruct_3d_basis Args: center: (N, L, 3), usually the position of C_alpha. p1: (N, L, 3), usually the position of C. p2: (N, L, 3)
evaluation/ddg/models/common.py:68