MTSAC implementation based on Metaworld. Benchmarked on MT50.

mtsac_metaworld_mt50(ctxt=None, *, seed, use_gpu, _gpu, n_tasks, timesteps)

Train MTSAC with MT50 environment.

  • 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.

  • use_gpu (bool) – Used to enable ussage of GPU in training.

  • _gpu (int) – The ID of the gpu (used on multi-gpu machines).

  • n_tasks (int) – Number of tasks to use. Should be a multiple of 50.

  • timesteps (int) – Number of timesteps to run.