AF3 make_data_config函数解读

AlphaFold3 feature_pipeline 模块 make_data_config 函数用于构造 AlphaFold3 的数据配置，并返回 更新后的配置和所需的特征名称列表。它主要用于 不同模式（如训练、推理）下的数据处理，确保特征选择符合设定，同时设置 裁剪尺寸（crop_size） 以匹配序列长度 num_res。

源代码：

def make_data_config(
    config: ml_collections.ConfigDict,
    mode: str,
    num_res: int,
) -> Tuple[ml_collections.ConfigDict, List[str]]:
    cfg = copy.deepcopy(config)
    mode_cfg = cfg[mode]
    # with cfg.unlocked():
    if mode_cfg.crop_size is None:
        mode_cfg.crop_size = num_res

    feature_names = cfg.common.unsupervised_features

    # Add seqemb related features if using seqemb mode.
    if cfg.seqemb_mode.enabled:
        feature_names += cfg.common.seqemb_features

    if cfg.common.use_templates:
        feature_names += cfg.common.template_features

    if cfg[mode].