Permalink
Branch: master
Find file Copy path
450f0f7 Apr 24, 2016
1 contributor

Users who have contributed to this file

22 lines (18 sloc) 646 Bytes
from 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)