Skip to content

Commit 8fd658e

Browse files
committed
fix: async ppo runner
Signed-off-by: Hao Lin <linhaomails@gmail.com>
1 parent 4e4daa7 commit 8fd658e

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

rlinf/runners/async_ppo_embodied_runner.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -188,39 +188,39 @@ def run(self) -> None:
188188
metrics_list=training_metrics,
189189
step=self.global_step,
190190
prefix="train",
191-
worker_group_name=self.actor_group_name,
191+
worker_group_name=self.actor.worker_group_name,
192192
)
193193
self._log_ranked_metrics(
194194
metrics_list=actor_time_metrics_per_rank,
195195
step=self.global_step,
196196
prefix="time/actor",
197-
worker_group_name=self.actor_group_name,
197+
worker_group_name=self.actor.worker_group_name,
198198
)
199199
self._log_ranked_metrics(
200200
metrics_list=rollout_metrics_list,
201201
step=self.global_step,
202202
prefix="rollout",
203-
worker_group_name=self.actor_group_name,
203+
worker_group_name=self.actor.worker_group_name,
204204
)
205205
self._log_ranked_metrics(
206206
metrics_list=env_time_metrics_per_rank,
207207
step=self.global_step,
208208
prefix="time/env",
209-
worker_group_name=self.env_group_name,
209+
worker_group_name=self.env.worker_group_name,
210210
add_prefix=False,
211211
)
212212
self._log_ranked_metrics(
213213
metrics_list=env_metrics_per_rank,
214214
step=self.global_step,
215215
prefix="env",
216-
worker_group_name=self.env_group_name,
216+
worker_group_name=self.env.worker_group_name,
217217
add_prefix=False,
218218
)
219219
self._log_ranked_metrics(
220220
metrics_list=rollout_time_metrics_per_rank,
221221
step=self.global_step,
222222
prefix="time/rollout",
223-
worker_group_name=self.rollout_group_name,
223+
worker_group_name=self.rollout.worker_group_name,
224224
add_prefix=False,
225225
)
226226

0 commit comments

Comments
 (0)