fix save issue

2025-06-01 20:05:27 +00:00 · 2025-04-27 17:51:47 +08:00 · 2025-04-27 17:51:47 +08:00 · 26d9eaddbc
commit 26d9eaddbc
parent 38008858e4
1 changed files with 1 additions and 0 deletions
--- a/applications/ColossalChat/coati/distributed/grpo_consumer.py
+++ b/applications/ColossalChat/coati/distributed/grpo_consumer.py
@ -62,6 +62,7 @@ class GRPOConsumer(BaseConsumer):
            model_config,
            plugin_config,
            minibatch_size,
+            save_interval=save_interval,
            save_dir=save_dir,
        )
        path = model_config.pop("path")