0d3ad1aad015bf2e705448c35b237937dd6cf739,tensorforce/models/model.py,Model,tf_initialize,#Model#,772

Before Change

            )
            self.global_episode = tf.get_variable(
                name="global-episode",
                dtype=util.tf_dtype("int"),
                trainable=False,
                initializer=0,
                collections=["global-episode"]

After Change

            // Global timestep
            collection = self.graph.get_collection(name="global-timestep")
            if len(collection) == 0:
                self.global_timestep = tf.get_variable(
                    name="global-timestep",
                    shape=(),
                    dtype=tf.int64,
                    trainable=False,
                    initializer=tf.constant_initializer(value=0, dtype=tf.int64),
                    collections=["global-timestep", tf.GraphKeys.GLOBAL_STEP]
                )
            else:
                assert len(collection) == 1
                self.global_timestep = collection[0]

            // Global episode
            collection = self.graph.get_collection(name="global-episode")
            if len(collection) == 0:
                self.global_episode = tf.get_variable(
                    name="global-episode",

Instances