Skip to content

Commit 28795f5

Browse files
TongLi3701Tong Li
andauthored
fix save issue (#6279)
Co-authored-by: Tong Li <[email protected]>
1 parent 3800885 commit 28795f5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

applications/ColossalChat/coati/distributed/grpo_consumer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ def __init__(
6262
model_config,
6363
plugin_config,
6464
minibatch_size,
65+
save_interval=save_interval,
6566
save_dir=save_dir,
6667
)
6768
path = model_config.pop("path")

0 commit comments

Comments
 (0)