4806717d2b2b9110b345e57a6565424609bee134,src/garage/tf/policies/gaussian_mlp_task_embedding_policy.py,GaussianMLPTaskEmbeddingPolicy,_initialize,#GaussianMLPTaskEmbeddingPolicy#,119

Before Change

            with tf.compat.v1.variable_scope("concat_obs_latent"):
                obs_latent_input = tf.concat([obs_input, latent_input], -1)
            self._dist, _, _ = self.model.build(obs_latent_input,
                                                name="given_latent")

            with tf.compat.v1.variable_scope("concat_obs_latent_var"):
                embed_state_input = tf.concat([obs_input, latent_var], -1)

            dist_given_task, _, _ = self.model.build(embed_state_input,
                                                     name="given_task")

        self._f_dist_obs_latent = tf.compat.v1.get_default_session(
        ).make_callable(
            [self._dist.sample(), self._dist.loc,
             self._dist.stddev()],
            feed_list=[obs_input, latent_input])

        self._f_dist_obs_task = tf.compat.v1.get_default_session(

After Change

            dist_given_task, mean_g_t, log_std_g_t = self.model.build(
                embed_state_input, name="given_task")

        self._f_dist_obs_latent = tf.compat.v1.get_default_session(
        ).make_callable([self._dist.sample(), mean_var, log_std_var],
                        feed_list=[obs_input, latent_input])

        self._f_dist_obs_task = tf.compat.v1.get_default_session(

Instances