73b21544b0feda16d2e8b69a1fe502aa387b8f89,tensorforce/core/optimizers/tf_optimizer.py,TFOptimizer,get_variables,#TFOptimizer#Any#Any#,125

Before Change

        for variable in self.optimizer.weights:
            name_index = variable.name.rindex("/" + self.name + "/")
            name = variable.name[name_index + len(self.name) + 2: -2]
            if name in self.variables:
                break
            self.variables[name] = variable
        for name, value in self.optimizer._hyper.items():
            if isinstance(value, tf.Variable):
                if name in self.variables:

After Change

    def get_variables(self, only_trainable=False, only_saved=False):
        optimizer = self.optimizer
        while True:
            for variable in optimizer.weights:
                name = "/" + self.name + "/"
                if name in variable.name:
                    name = variable.name[variable.name.rindex(name) + len(name): -2]
                else:
                    name = variable.name[variable.name.rindex("/") + 1: -2]
                self.variables[name] = variable
            for name, value in optimizer._hyper.items():
                if isinstance(value, tf.Variable):
                    self.variables[name] = value
            if hasattr(optimizer, "_ema"):
                for variable in optimizer._ema._averages.values():
                    assert variable.name.startswith("agent/") and \
                        variable.name.endswith("/ExponentialMovingAverage:0")
                    self.variables[variable.name[:-2]] = variable
            if hasattr(optimizer, "_optimizer"):
                optimizer = optimizer._optimizer
            else:
                break

        variables = super().get_variables(only_trainable=only_trainable, only_saved=only_saved)

        return variables

Instances