↓ 1 callersMethodcoord_model coord: [bs * n_node, n_channel, d] edge_index: list of [n_edge], [n_edge] coord_diff: [n_edge, n_channel, d] edge_fea
models/MCAttGNN/mc_egnn.py:107
↓ 1 callersMethodforward_beta Args: R: Frame basis matrices, (N, L, 3, 3_index). t: Frame external (absolute) coordinates, (N, L, 3).
evaluation/ddg/models/attention.py:172
↓ 1 callersMethodforward_vpoint Args: R: Frame basis matrices, (N, L, 3, 3_index). t: Frame external (absolute) coordinates, (N, L, 3).
evaluation/ddg/models/attention.py:203