garage.tf.plotter

class Plotter(env, policy, sess=None, graph=None, rollout=default_rollout)

Visualizes episodes of the policy as it trains.

Usually, this class is used by sending plot=True to LocalRunner.train().

Parameters:
  • env (gym.Env) – Environment from which to visualize episodes. This will
  • used without copying in the current process but in a separate (be) –
  • so it should be given a unique copy (in particular, do not pass (thread,) –
  • training environment here, then try to pickle it, or you will (the) –
  • get crashes) (occasionally) –
  • policy (garage.tf.Policy) – Policy used to visualize episodes.
  • sess (tf.Session) – The TensorFlow session to use.
  • graph (tf.Graph) – The TensorFlow graph to use.
  • rollout (callable) – The rollout function to call.
close(self)

Stop the Plotter’s worker thread.

static get_plotters()

Return all garage.tf.Plotter’s.

Returns:All the garage.tf.Plotter’s
Return type:list[garage.tf.Plotter]
start(self)

Start the Plotter’s worker thread.

update_plot(self, policy, max_length=np.inf)

Update the policy being plotted.

Parameters:
  • policy (garage.tf.Policy) – Policy to visualize.
  • max_length (int or float) – The maximum length to allow an episode to be. Defaults to infinity.