Skip to content

Commit 0590f10

Browse files
author
Tong Li
committed
update select algo
1 parent 0cc0c84 commit 0590f10

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

applications/ColossalChat/rl_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
parser.add_argument("-tbs", "--train-batch-size", type=int, default=16)
1616
parser.add_argument("-tmbs", "--train-microbatch-size", type=int, default=2)
1717
parser.add_argument("-b", "--backend", type=str, default="transformers")
18-
parser.add_argument("-a", "--algo", type=str, default="GPRO", choices=["Simple, GPRO"])
18+
parser.add_argument("-a", "--algo", type=str, default="GRPO", choices=["Simple, GPRO"])
1919
args = parser.parse_args()
2020

2121
ray.init(address="local", namespace="ray-example")

0 commit comments

Comments
 (0)