Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions examples/configs/grpo_math_1B.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -496,6 +496,10 @@ data_plane:
storage_capacity: 1000000 # max samples retained per partition
num_storage_units: 2 # storage shards
claim_meta_poll_interval_s: 0.5 # blocking-claim poll cadence
# SingleController only: save native TQ state. Required when trainer
# checkpointing is enabled with a replay-checkpoint-capable sampler;
# supported samplers restore from metadata-only replay indexes.
checkpointing_enabled: false
global_segment_size: 549755813888 # 512 GiB — used when backend == "mooncake_cpu"
local_buffer_size: 68719476736 # 64 GiB — used when backend == "mooncake_cpu"
# observability: # NotRequired
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@ async_rl:
# Enable per-rollout diagnostic prints (prompt content / completion previews).
diagnostics: false

# SC does not support checkpointing yet.
checkpointing:
enabled: false
checkpoint_dir: results/grpo-single-controller
Expand Down Expand Up @@ -55,6 +54,8 @@ logger:

data_plane:
enabled: true
# Required shadow snapshot: a save failure aborts checkpoint finalization.
checkpointing_enabled: true

cluster:
gpus_per_node: 2
Expand Down
Loading
Loading