garage.sampler.batch_sampler module

Class with batch-based sampling.

class BatchSampler(algo, env)[source]

Bases: garage.sampler.base.BaseSampler

Class with batch-based sampling.

Parameters:
  • algo (garage.np.algos.RLAlgorithm) – The algorithm.
  • env (gym.Env) – The environment.
obtain_samples(itr, batch_size=None, whole_paths=True)[source]

Obtain samples.

shutdown_worker()[source]

Shutdown workers.

start_worker()[source]

Start workers.