Skip to content

Commit 68989fe

Browse files
authored
fix(embody): create channel in runner (RLinf#453)
Signed-off-by: Lin-xs <1833080950@qq.com>
1 parent ba73030 commit 68989fe

File tree

60 files changed

+105
-809
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

60 files changed

+105
-809
lines changed

examples/embodiment/config/behavior_openvlaoft_eval.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,6 @@ algorithm:
8181

8282
env:
8383
group_name: "EnvGroup"
84-
channel:
85-
name: "env_buffer_list"
86-
queue_name: "obs_buffer"
87-
queue_size: 0
8884
enable_offload: False
8985

9086
# Override the default values in env/behavior_r1pro
@@ -98,11 +94,6 @@ env:
9894

9995
rollout:
10096
group_name: "RolloutGroup"
101-
channel:
102-
name: ${env.channel.name}
103-
queue_name: "action_buffer"
104-
queue_size: 0
105-
mode: "colocate"
10697
backend: "huggingface"
10798
enable_offload: True
10899
pipeline_stage_num: 1
@@ -113,10 +104,6 @@ rollout:
113104

114105
actor:
115106
group_name: "ActorGroup"
116-
channel:
117-
name: ${env.channel.name}
118-
queue_name: "replay_buffer"
119-
queue_size: 0
120107
training_backend: "fsdp"
121108
micro_batch_size: 2
122109
global_batch_size: 160

examples/embodiment/config/behavior_ppo_openvlaoft.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -83,10 +83,6 @@ algorithm:
8383

8484
env:
8585
group_name: "EnvGroup"
86-
channel:
87-
name: "env_buffer_list"
88-
queue_name: "obs_buffer"
89-
queue_size: 0
9086
enable_offload: False
9187

9288
# Override the default values in env/behavior_r1pro
@@ -104,11 +100,6 @@ env:
104100

105101
rollout:
106102
group_name: "RolloutGroup"
107-
channel:
108-
name: ${env.channel.name}
109-
queue_name: "action_buffer"
110-
queue_size: 0
111-
mode: "colocate"
112103
backend: "huggingface"
113104
enable_offload: True
114105
pipeline_stage_num: 1
@@ -119,10 +110,6 @@ rollout:
119110

120111
actor:
121112
group_name: "ActorGroup"
122-
channel:
123-
name: ${env.channel.name}
124-
queue_name: "replay_buffer"
125-
queue_size: 0
126113
training_backend: "fsdp"
127114
micro_batch_size: 2
128115
global_batch_size: 160

examples/embodiment/config/calvin_d_d_ppo_openpi.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -84,10 +84,6 @@ algorithm:
8484

8585
env:
8686
group_name: "EnvGroup"
87-
channel:
88-
name: "env_buffer_list"
89-
queue_name: "obs_buffer"
90-
queue_size: 0
9187
enable_offload: False
9288

9389
train:
@@ -110,11 +106,6 @@ env:
110106

111107
rollout:
112108
group_name: "RolloutGroup"
113-
channel:
114-
name: ${env.channel.name}
115-
queue_name: "action_buffer"
116-
queue_size: 0
117-
mode: "colocate"
118109
generation_backend: "huggingface"
119110
recompute_logprobs: False
120111
enable_offload: False
@@ -126,10 +117,6 @@ rollout:
126117

127118
actor:
128119
group_name: "ActorGroup"
129-
channel:
130-
name: ${env.channel.name}
131-
queue_name: "replay_buffer"
132-
queue_size: 0
133120
training_backend: "fsdp"
134121
micro_batch_size: 128
135122
global_batch_size: 2048

examples/embodiment/config/calvin_d_d_ppo_openpi_pi05.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -84,10 +84,6 @@ algorithm:
8484

8585
env:
8686
group_name: "EnvGroup"
87-
channel:
88-
name: "env_buffer_list"
89-
queue_name: "obs_buffer"
90-
queue_size: 0
9187
enable_offload: False
9288

9389
train:
@@ -115,11 +111,6 @@ env:
115111

116112
rollout:
117113
group_name: "RolloutGroup"
118-
channel:
119-
name: ${env.channel.name}
120-
queue_name: "action_buffer"
121-
queue_size: 0
122-
mode: "colocate"
123114
generation_backend: "huggingface"
124115
recompute_logprobs: False
125116
enable_offload: False
@@ -131,10 +122,6 @@ rollout:
131122

132123
actor:
133124
group_name: "ActorGroup"
134-
channel:
135-
name: ${env.channel.name}
136-
queue_name: "replay_buffer"
137-
queue_size: 0
138125
training_backend: "fsdp"
139126

140127
micro_batch_size: 128

examples/embodiment/config/isaaclab_ppo_gr00t_demo.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -88,10 +88,6 @@ algorithm:
8888

8989
env:
9090
group_name: "EnvGroup"
91-
channel:
92-
name: "env_buffer_list"
93-
queue_name: "obs_buffer"
94-
queue_size: 0
9591
enable_offload: False
9692

9793
# Override the default values in env/isaaclab_stack_cube
@@ -115,11 +111,6 @@ env:
115111

116112
rollout:
117113
group_name: "RolloutGroup"
118-
channel:
119-
name: ${env.channel.name}
120-
queue_name: "action_buffer"
121-
queue_size: 0
122-
mode: "colocate"
123114
backend: "huggingface"
124115
enable_offload: True
125116
pipeline_stage_num: 1
@@ -130,10 +121,6 @@ rollout:
130121

131122
actor:
132123
group_name: "ActorGroup"
133-
channel:
134-
name: ${env.channel.name}
135-
queue_name: "replay_buffer"
136-
queue_size: 0
137124
training_backend: "fsdp"
138125
micro_batch_size: 1
139126
global_batch_size: 2

examples/embodiment/config/libero_10_grpo_openpi.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,6 @@ algorithm:
8686

8787
env:
8888
group_name: "EnvGroup"
89-
channel:
90-
name: "env_buffer_list"
91-
queue_name: "obs_buffer"
92-
queue_size: 0
9389
enable_offload: False
9490

9591
train:
@@ -112,11 +108,6 @@ env:
112108

113109
rollout:
114110
group_name: "RolloutGroup"
115-
channel:
116-
name: ${env.channel.name}
117-
queue_name: "action_buffer"
118-
queue_size: 0
119-
mode: "colocate"
120111
generation_backend: "huggingface"
121112
recompute_logprobs: False
122113
unnorm_key: libero_10
@@ -129,10 +120,6 @@ rollout:
129120

130121
actor:
131122
group_name: "ActorGroup"
132-
channel:
133-
name: ${env.channel.name}
134-
queue_name: "replay_buffer"
135-
queue_size: 0
136123
training_backend: "fsdp"
137124
micro_batch_size: 128
138125
global_batch_size: 2048 # maximum : global batch size = micro batch size * actor num_processes.

examples/embodiment/config/libero_10_grpo_openpi_pi05.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -85,10 +85,6 @@ algorithm:
8585

8686
env:
8787
group_name: "EnvGroup"
88-
channel:
89-
name: "env_buffer_list"
90-
queue_name: "obs_buffer"
91-
queue_size: 0
9288
enable_offload: False
9389

9490
train:
@@ -112,11 +108,6 @@ env:
112108

113109
rollout:
114110
group_name: "RolloutGroup"
115-
channel:
116-
name: ${env.channel.name}
117-
queue_name: "action_buffer"
118-
queue_size: 0
119-
mode: "colocate"
120111
generation_backend: "huggingface"
121112
recompute_logprobs: False
122113
unnorm_key: libero_10
@@ -129,10 +120,6 @@ rollout:
129120

130121
actor:
131122
group_name: "ActorGroup"
132-
channel:
133-
name: ${env.channel.name}
134-
queue_name: "replay_buffer"
135-
queue_size: 0
136123
training_backend: "fsdp"
137124
micro_batch_size: 128
138125
global_batch_size: 2048 # maximum : global batch size = micro batch size * actor num_processes.

examples/embodiment/config/libero_10_grpo_openvlaoft.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,6 @@ algorithm:
8686

8787
env:
8888
group_name: "EnvGroup"
89-
channel:
90-
name: "env_buffer_list"
91-
queue_name: "obs_buffer"
92-
queue_size: 0
9389
enable_offload: False
9490

9591
# Override the default values in env/libero_10
@@ -115,11 +111,6 @@ env:
115111

116112
rollout:
117113
group_name: "RolloutGroup"
118-
channel:
119-
name: ${env.channel.name}
120-
queue_name: "action_buffer"
121-
queue_size: 0
122-
mode: "colocate"
123114
generation_backend: "huggingface"
124115
enable_offload: False
125116
pipeline_stage_num: 1
@@ -130,10 +121,6 @@ rollout:
130121

131122
actor:
132123
group_name: "ActorGroup"
133-
channel:
134-
name: ${env.channel.name}
135-
queue_name: "replay_buffer"
136-
queue_size: 0
137124
training_backend: "fsdp"
138125
micro_batch_size: 32
139126
global_batch_size: 16384

examples/embodiment/config/libero_10_grpo_openvlaoft_eval.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,6 @@ algorithm:
8686

8787
env:
8888
group_name: "EnvGroup"
89-
channel:
90-
name: "env_buffer_list"
91-
queue_name: "obs_buffer"
92-
queue_size: 0
9389
enable_offload: False
9490

9591
eval:
@@ -109,11 +105,6 @@ env:
109105

110106
rollout:
111107
group_name: "RolloutGroup"
112-
channel:
113-
name: ${env.channel.name}
114-
queue_name: "action_buffer"
115-
queue_size: 0
116-
mode: "colocate"
117108
generation_backend: "huggingface"
118109
enable_offload: False
119110
pipeline_stage_num: 1
@@ -124,10 +115,6 @@ rollout:
124115

125116
actor:
126117
group_name: "ActorGroup"
127-
channel:
128-
name: ${env.channel.name}
129-
queue_name: "replay_buffer"
130-
queue_size: 0
131118
training_backend: "fsdp"
132119
micro_batch_size: 32
133120
global_batch_size: 8192

examples/embodiment/config/libero_10_ppo_gr00t.yaml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -83,10 +83,6 @@ algorithm:
8383

8484
env:
8585
group_name: "EnvGroup"
86-
channel:
87-
name: "env_buffer_list"
88-
queue_name: "obs_buffer"
89-
queue_size: 0
9086
enable_offload: False
9187
train:
9288
total_num_envs: 64
@@ -110,11 +106,6 @@ env:
110106

111107
rollout:
112108
group_name: "RolloutGroup"
113-
channel:
114-
name: ${env.channel.name}
115-
queue_name: "action_buffer"
116-
queue_size: 0
117-
mode: "colocate"
118109
backend: "huggingface"
119110
enable_offload: True
120111
pipeline_stage_num: 1
@@ -125,10 +116,6 @@ rollout:
125116

126117
actor:
127118
group_name: "ActorGroup"
128-
channel:
129-
name: ${env.channel.name}
130-
queue_name: "replay_buffer"
131-
queue_size: 0
132119
training_backend: "fsdp"
133120
micro_batch_size: 128
134121
global_batch_size: 1024

0 commit comments

Comments
 (0)