Added timeout cli.

Author: Miguel Alonso Jr

ml-agents/mlagents/trainers/cli_utils.py

@@ -233,6 +233,13 @@ def _create_parser() -> argparse.ArgumentParser:
         help="Results base directory",
     )
 
+    argparser.add_argument(
+        "--timeout-wait",
+        default=60,
+        help="The period of time to wait on a Unity environment to startup for training.",
+        action=DetectDefault,
+    )
+
     eng_conf = argparser.add_argument_group(title="Engine Configuration")
     eng_conf.add_argument(
         "--width",

ml-agents/mlagents/trainers/learn.py

@@ -101,6 +101,7 @@ def run_training(run_seed: int, options: RunOptions, num_areas: int) -> None:
             engine_settings.no_graphics,
             run_seed,
             num_areas,
+            env_settings.timeout_wait,
             port,
             env_settings.env_args,
             os.path.abspath(run_logs_dir),  # Unity environment requires absolute path
@@ -175,6 +176,7 @@ def create_environment_factory(
     no_graphics: bool,
     seed: int,
     num_areas: int,
+    timeout_wait: int,
     start_port: Optional[int],
     env_args: Optional[List[str]],
     log_folder: str,
@@ -194,6 +196,7 @@ def create_unity_environment(
             additional_args=env_args,
             side_channels=side_channels,
             log_folder=log_folder,
+            timeout_wait=timeout_wait,
         )
 
     return create_unity_environment

ml-agents/mlagents/trainers/settings.py

@@ -817,6 +817,7 @@ class EnvironmentSettings:
     base_port: int = parser.get_default("base_port")
     num_envs: int = attr.ib(default=parser.get_default("num_envs"))
     num_areas: int = attr.ib(default=parser.get_default("num_areas"))
+    timeout_wait: int = attr.ib(default=parser.get_default("timeout_wait"))
     seed: int = parser.get_default("seed")
     max_lifetime_restarts: int = parser.get_default("max_lifetime_restarts")
     restarts_rate_limit_n: int = parser.get_default("restarts_rate_limit_n")