MCPcopy Index your code
hub / github.com/OpenDriveLab/DriveAdapter / get_avg_ep_stat

Method get_avg_ep_stat

roach/utils/wandb_callback.py:195–211  ·  view source on GitHub ↗
(ep_stat_buffer, prefix='')

Source from the content-addressed store, hash-verified

193
194 @staticmethod
195 def get_avg_ep_stat(ep_stat_buffer, prefix=''):
196 avg_ep_stat = {}
197 if len(ep_stat_buffer) > 0:
198 for ep_info in ep_stat_buffer:
199 for k, v in ep_info.items():
200 k_avg = f'{prefix}{k}'
201 if k_avg in avg_ep_stat:
202 avg_ep_stat[k_avg] += v
203 else:
204 avg_ep_stat[k_avg] = v
205
206 n_episodes = float(len(ep_stat_buffer))
207 for k in avg_ep_stat.keys():
208 avg_ep_stat[k] /= n_episodes
209 avg_ep_stat[f'{prefix}n_episodes'] = n_episodes
210
211 return avg_ep_stat

Callers 2

_on_rollout_endMethod · 0.95
evaluate_policyMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected