提交 ee97ec62 编写于 作者: B BigFishMaster

Add memory optimization amd nccl

上级 a8d66cc0
...@@ -270,7 +270,6 @@ if __name__ == '__main__': ...@@ -270,7 +270,6 @@ if __name__ == '__main__':
lr_strategy = {"bd": bd, "lr": lr} lr_strategy = {"bd": bd, "lr": lr}
use_nccl = True use_nccl = True
# layers: 50, 152 # layers: 50, 152
layers = 50 layers = 50
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册