d19e9ab3b446e595c72e03fbc03684927d77526b,src/garage/tf/policies/categorical_mlp_policy.py,CategoricalMLPPolicy,dist_info_sym,#CategoricalMLPPolicy#Any#Any#Any#,73

Before Change

    @overrides
    def dist_info_sym(self, obs_var, state_info_vars=None, name=None):
        with tf.name_scope(name, "dist_info_sym", [obs_var, state_info_vars]):
            with tf.name_scope(self._prob_network_name, values=[obs_var]):
                prob = L.get_output(
                    self._l_prob, {self._l_obs: tf.cast(obs_var, tf.float32)})
            return dict(prob=prob)

    @overrides
    def dist_info(self, obs, state_infos=None):
        return dict(prob=self._f_prob(obs))

After Change

    @overrides
    def dist_info_sym(self, obs_var, state_info_vars=None, name=None):
        Symbolic graph of the distribution.
        with tf.compat.v1.variable_scope(self._variable_scope):
            prob = self.model.build(obs_var, name=name)
        return dict(prob=prob)

    @overrides
    def dist_info(self, obs, state_infos=None):

Instances