050f278d77b1a4982c14285fdc2c780d57d31194,tests/garage/tf/models/test_gaussian_mlp_model.py,TestGaussianMLPModel,test_without_std_share_network_is_pickleable,#TestGaussianMLPModel#Any#Any#Any#,205

Before Change

        output1 = self.sess.run(outputs[:-1], feed_dict={input_var: self.obs})
        with tf.Session(graph=tf.Graph()) as sess:
            input_var = tf.placeholder(tf.float32, shape=(None, 5))
            model_pickled = pickle.loads(pickle.dumps(model))
            outputs = model_pickled.build(input_var)
            output2 = sess.run(outputs[:-1], feed_dict={input_var: self.obs})
            assert np.array_equal(output1, output2)

After Change

        outputs = model.build(input_var)

        // get output bias
        with tf.variable_scope("GaussianMLPModel", reuse=True):
            bias = tf.get_variable("dist_params/mean_network/output/bias")
        // assign it to all ones
        self.sess.run(tf.assign(bias, tf.ones_like(bias)))

        output1 = self.sess.run(outputs[:-1], feed_dict={input_var: self.obs})

        h = pickle.dumps(model)

Instances