# MEGAFold-Multimer

**Repository Path**: wangmin0104/MEGAFold-Multimer

## Basic Information

- **Project Name**: MEGAFold-Multimer
- **Description**: MEGAFold-Multimer
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2026-01-15
- **Last Updated**: 2026-01-21

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# MEGAFold-Multimer
基于昇思 MindSpore 深度学习框架与昇腾 AI 硬件平台深度协同优化，研发完成的蛋白质复合物结构预测模型，具备多场景的结构预测能力 —— 不仅可高效完成常规复合物蛋白质的结构预测，还能精准实现带约束条件的复合物结构预测，这一功能可通过推理脚本中是否传入约束参数灵活控制：当输入特定约束参数时，模型将依据预设的约束条件预测复合物结构；未输入约束参数时，则默认执行无约束的结构预测。


本版模型重点面向昇腾 910B AI 处理器（配备约 64GB 高带宽显存）完成了适配，大幅提升了单卡推理的序列长度上限，目前单卡可稳定支持序列长度达 3072 的复合物蛋白质结构推理，相较于传统方案，在无需多卡集群的前提下，即可满足更长序列、更复杂构象的预测需求，有效降低了大尺度蛋白质复合物结构研究的硬件门槛和计算成本。


### 硬件支持情况

| 硬件平台      | 操作系统        | 状态 |
| :------------ | :-------------- | :--- |
| Ascend 910B    | CentOS-aarch64  | ✔️ |


### 软件版本
依赖配套如下表

| 依赖软件         |                                                              |
| ---------------- | ------------------------------------------------------------ |
| MindSpore        | [MindSpore 1.9.0](https://www.mindspore.cn/install/)         |
| MindSPONGE       | 0.6.0                                                     |
| Python           | >=3.7                                           |

#### 使用说明

MEGAFold-Multimer推理启动命令

```bash

bash infer_main.sh ${length} "H1106.pkl;1;20;0;./output/;{ckpt_path}" log.log
```

#### 参与贡献

1.  Fork 本仓库
2.  新建 Feat_xxx 分支
3.  提交代码
4.  新建 Pull Request