From 76c7c397549e7ee2fb43d3d6b9e49e124c2e81c7 Mon Sep 17 00:00:00 2001 From: Xiaoxiang Zhang Date: Tue, 22 Feb 2022 12:48:16 -0800 Subject: [PATCH] add log performance of each episode Summary: 1. add log performance of each episode 2. crease usecase specific episode post callback 3. create step post callback Reviewed By: vgup0, alexnikulkov Differential Revision: D34295015 fbshipit-source-id: 630b499313034319da34057de5a28e8972496b5d --- reagent/gym/datasets/replay_buffer_dataset.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/reagent/gym/datasets/replay_buffer_dataset.py b/reagent/gym/datasets/replay_buffer_dataset.py index 053aeb2ba..eea3c713b 100644 --- a/reagent/gym/datasets/replay_buffer_dataset.py +++ b/reagent/gym/datasets/replay_buffer_dataset.py @@ -134,6 +134,8 @@ def __iter__(self): possible_actions_mask = next_possible_actions_mask num_steps += 1 global_num_steps += 1 + if self._agent.post_step: + self._agent.post_step(transition) if self._post_episode_callback: self._post_episode_callback(trajectory, info)