Permalink
Cannot retrieve contributors at this time
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 up
Fetching contributors…

from rllab.algos.npo import NPO | |
from rllab.optimizers.conjugate_gradient_optimizer import ConjugateGradientOptimizer | |
from rllab.core.serializable import Serializable | |
class TRPO(NPO): | |
""" | |
Trust Region Policy Optimization | |
""" | |
def __init__( | |
self, | |
optimizer=None, | |
optimizer_args=None, | |
**kwargs): | |
if optimizer is None: | |
if optimizer_args is None: | |
optimizer_args = dict() | |
optimizer = ConjugateGradientOptimizer(**optimizer_args) | |
super(TRPO, self).__init__(optimizer=optimizer, **kwargs) |