a41f2ff4c1437f0b61e76265c31bdec71be0556f,softlearning/algorithms/sac.py,SAC,get_diagnostics,#SAC#Any#Any#Any#Any#,358

Before Change

                 self.global_step),
                feed_dict))

        diagnostics = OrderedDict({
            "Q-avg": np.mean(Q_values),
            "Q-std": np.std(Q_values),
            "Q_loss": np.mean(Q_losses),
            "policy_loss": np.mean(policy_losses),
            "alpha": alpha,
        })

        policy_diagnostics = self._policy.get_diagnostics(
            batch["observations"])
        diagnostics.update({
            f"policy/{key}": value
            for key, value in policy_diagnostics.items()
        })

        if self._plotter:
            self._plotter.draw()

After Change

        diagnostics.update(OrderedDict([
            (f"policy/{key}", value)
            for key, value in
            self._policy.get_diagnostics(batch["observations"]).items()
        ]))

        if self._plotter:
            self._plotter.draw()

Instances