RuntimeWarning: 在 reduce 中遇到溢出
作者: wtni-gidle创建于 2024年12月26日更新于 2026年8月17日
标签question
我一直在使用 AlphaFold3 预测一些相对较大的蛋白质。我惊讶地发现，使用 A40 (48GB)，AlphaFold3 可以成功预测一些长度超过 6000 倍的蛋白质！当然，这需要启用统一内存并设置 `pair_transition_shard_spec`。然而，在特征提取过程中，有时会出现以下警告：
内容来源: google-deepmind/alphafold3