想问一下训练资源以及时间

dptech-corp / Uni-Fold

An open-source platform for developing protein models beyond AlphaFold.

https://doi.org/10.1101/2022.08.04.502811

Apache License 2.0

380 stars 74 forks source link

Closed onlyonewater closed 1 year ago

onlyonewater commented 1 year ago

利用这个代码，我手上有8张A100 (80G)，从头开始训练alphafold-multimer大概需要多久呢？

ZiyaoLi commented 1 year ago

可以按照https://www.biorxiv.org/content/10.1101/2022.08.04.502811v3.full.pdf table4里128卡时间数线性计算，即乘以16倍。

onlyonewater commented 1 year ago

好的，谢谢

onlyonewater commented 1 year ago

想问一下，作者，你们这个代码可以在3090上训练嘛？如果有8张3090，每张卡上的batch size是1，在训练的时候会有out of memory现象出现嘛？

onlyonewater commented 1 year ago

@ZiyaoLi

ZiyaoLi commented 1 year ago

我没有试过。经验上显存占用小于16G。

onlyonewater commented 1 year ago

好的，多谢啦！！！