Permalink
Join GitHub today
GitHub is home to over 31 million developers working together to host and review code, manage projects, and build software together.
Sign upfrom rllab.optimizers.penalty_lbfgs_optimizer import PenaltyLbfgsOptimizer | |
from rllab.algos.npo import NPO | |
from rllab.core.serializable import Serializable | |
class PPO(NPO, Serializable): | |
""" | |
Penalized Policy Optimization. | |
""" | |
def __init__( | |
self, | |
optimizer=None, | |
optimizer_args=None, | |
**kwargs): | |
Serializable.quick_init(self, locals()) | |
if optimizer is None: | |
if optimizer_args is None: | |
optimizer_args = dict() | |
optimizer = PenaltyLbfgsOptimizer(**optimizer_args) | |
super(PPO, self).__init__(optimizer=optimizer, **kwargs) |