ouyangshixiong / xparl_qmix Goto Github PK

English Version:

• QMIX algorithm implementation using PARL framework of PaddlePaddle
• Original algorithm is from PARL official example
• Add XPARL support for training on muitiple CPUs，in theory training speed will be much faster.

• setup ubuntu
• install PaddlePaddle 2.x+ version
• install PARL 2.x+ version
• install opencv-python
• install SC2
• install smac&Maps
• pull this code
• start xparl using command line: xparl start --port 8010 --cpu_num x
• start training using command line: python train.py

中文版：

• 用百度飞桨的PARL框架实现了QMIX算法
• 代码是基于百度飞桨PARL的example目录中的原QMIX算法改进而来
• 实现了基于xparl集群的多CPU训练，理论上可以大大提升训练速度

• 安装ubuntu
• 安装飞桨2.x以上版本，最好是基于GPU的版本。
• 安装PARL2.x版本
• 安装opencv-python，以支持cv2
• 在本机安装SC2
• 安装smac环境和对应的训练地图，这个按smac的github指导安装和验证即可。
• 拉这份代码到本地
• 先启动xparl集群，开启分布式cpu训练：xparl start --port 8010 --cpu_num x 。其中x为需要启动的cpu数，需要跟实际cpu数目匹配。
• 在命令行执行： python train.py 开启训练。可以同步开启visualdl，查看训练效果。