8ef3f88cfe38c9178ec53cca4025eeacbe230741,catalyst/rl/core/sampler.py,ValidSampler,_run_sample_loop,#ValidSampler#,412

Before Change

            if self.logger is not None:
                self.logger.add_scalar(
                    "trajectory/_mean_valid_reward",
                    np.mean(trajectories_rewards),
                    self.db_server.epoch
                )
            self._log_to_wandb(

After Change

            self._log_to_wandb(step=self.db_server.epoch, **loop_metrics)

            self.checkpoint["reward"] = trajectories_reward
            self.checkpoint["raw_reward"] = trajectories_raw_reward
            self.checkpoint["epoch"] = self.db_server.epoch
            self.save_checkpoint(
                logdir=self.logdir,

Instances