ouzhang57's picture
Upload folder using huggingface_hub (part 8)
505b983 verified
|
Raw
History Blame Contribute Delete
677 Bytes
# dummy optimizer
## 问题分析
朴素的 pipeline parallel 实现中,不支持某个 pipeline stage 的 parameter 都不需要参数更新或不需要反向计算。
## 解决方案
创建空 tensor,规避 optimizer 中所有 parameter 都不需要更新的场景。
在 pipeline parallel 的反向前加判断,若没有 grad_fn 则不进行反向计算。
## 使用方法
1. 在模型入口脚本中导入 patch 模块(InternVL/Qwen2VL 已支持);
```python
from mindspeed_mm.patchs import dummy_optimizer_patch
```
2. 在模型启动 shell 中添加参数;
```shell
GPT_ARGS="
...
--enable-dummy-optimizer \
"
```