bdd1a023482b34aa9980c64b434ce75824e16543,sac/policies/gmm.py,GMMPolicy,get_action,#GMMPolicy#Any#,75

Before Change

            raise AttributeError

        // Get first the GMM means.
        feeds = {self._obs_pl: obs[None]}
        mus = tf.get_default_session().run(self._dist.mus_t, feeds)[0]  // K x Da

        qs = self._qf.eval(obs[None], mus)

        if self._fixed_h is not None:
            h = self._fixed_h
        else:
            h = np.argmax(qs)

        return mus[h, :], {}  // Da

    @contextmanager
    def fix_h(self, h):

After Change

        TODO: Modify `NNPolicy.get_action` and remove this
        
        return self.get_actions(observation[None])[0], {}

    @overrides
    def get_actions(self, obs):
        Sample actions based on the observations.

Instances