def optimizer(self): if optimizer_fn: optimizers = optimizer_fn(self, self.config) else: optimizers = TorchPolicy.optimizer(self) optimizers = force_list(optimizers) if getattr(self, "exploration", None): optimizers = self.exploration.get_exploration_optimizer( optimizers) return optimizers
def optimizer(self): if optimizer_fn: optimizers = optimizer_fn(self, self.config) else: optimizers = TorchPolicy.optimizer(self) optimizers = force_list(optimizers) if hasattr(self, "exploration"): exploration_optimizers = force_list( self.exploration.get_exploration_optimizer(self.config)) optimizers.extend(exploration_optimizers) return optimizers
def optimizer(self): if optimizer_fn: return optimizer_fn(self, self.config) else: return TorchPolicy.optimizer(self)