From 0590f10fb78f45e9b94c4a928379262b9bfad4e0 Mon Sep 17 00:00:00 2001 From: Tong Li Date: Thu, 6 Mar 2025 16:27:13 +0800 Subject: [PATCH] update select algo --- applications/ColossalChat/rl_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/applications/ColossalChat/rl_example.py b/applications/ColossalChat/rl_example.py index 40231582d..347ffeeae 100644 --- a/applications/ColossalChat/rl_example.py +++ b/applications/ColossalChat/rl_example.py @@ -15,7 +15,7 @@ if __name__ == "__main__": 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("-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() ray.init(address="local", namespace="ray-example")