github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-23 06:00:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

update select algo

Browse Source
This commit is contained in:
Tong Li 2025-03-06 16:27:13 +08:00
parent 0cc0c843ed
commit 0590f10fb7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ColossalChat/rl_example.py
View File

@ -15,7 +15,7 @@ if __name__ == "__main__":
parser.add_argument("-tbs", "--train-batch-size", type=int, default=16) parser.add_argument("-tbs", "--train-batch-size", type=int, default=16)
parser.add_argument("-tmbs", "--train-microbatch-size", type=int, default=2) parser.add_argument("-tmbs", "--train-microbatch-size", type=int, default=2)
parser.add_argument("-b", "--backend", type=str, default="transformers") parser.add_argument("-b", "--backend", type=str, default="transformers")
parser.add_argument("-a", "--algo", type=str, default="GPRO", choices=["Simple, GPRO"]) parser.add_argument("-a", "--algo", type=str, default="GRPO", choices=["Simple, GPRO"])
args = parser.parse_args() args = parser.parse_args()
ray.init(address="local", namespace="ray-example") ray.init(address="local", namespace="ray-example")