trpo_pendulum

This is an example to train a task with TRPO algorithm (PyTorch).

Here it runs InvertedDoublePendulum-v2 environment with 100 iterations.

trpo_pendulum(ctxt=None, seed=1)

Train TRPO with InvertedDoublePendulum-v2 environment.

Parameters

  • ctxt (garage.experiment.ExperimentContext) – The experiment configuration used by Trainer to create the snapshotter.

  • seed (int) – Used to seed the random number generator to produce determinism.