embodichain.learning.rl.train

embodichain.learning.rl.train#

Overview#

Training entry points and command-line helpers for launching RL experiments.

Functions

cli()

Command-line interface for RL training.

parse_args()

Parse command line arguments.

train_from_config(config_path[, distributed])

Run training from a config file path.

Functions:

cli()

Command-line interface for RL training.

parse_args()

Parse command line arguments.

train_from_config(config_path[, distributed])

Run training from a config file path.
embodichain.learning.rl.train.cli()[source]#

Command-line interface for RL training.

Parses CLI arguments and launches training from a config file.

Task packages are discovered (and init hooks executed) before training so that task environments registered in separate packages (e.g. embodichain_tasks) are available to build_env. This mirrors the run_env CLI.

Return type:

None

embodichain.learning.rl.train.parse_args()[source]#

Parse command line arguments.

embodichain.learning.rl.train.train_from_config(config_path, distributed=None)[source]#

Run training from a config file path.

Parameters:

  • config_path (str) – Path to the training config file (.json, .yaml, or .yml).

  • distributed (bool | None) – If True, run multi-GPU distributed training. If None, use trainer.distributed from config.