Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
wzzzd authored Aug 2, 2021
1 parent 36ed697 commit 2990b8b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ Mask Language Model,简称Mask LM,即基于Mask机制的预训练语言模
```

### 多卡模式
如果你足够幸运,拥有了多张GPU卡,那么恭喜你,你可以进入起飞模式。🚀🚀
如果你不那么幸运,拥有了多张GPU卡,那么恭喜你,你可以进入起飞模式。🚀🚀

(1)通过修改Config.py文件中的变量self.cuda_visible_devices,指定你需要在哪几张卡上运行,卡号之间以英文逗号隔开。
(2)因为使用的是accelerate加速模块,故需要配置一些环境信息,在确保安装完环境所以的依赖包后,运行命令:
Expand Down

0 comments on commit 2990b8b

Please sign in to comment.