浏览代码

Adding new command line arguments

/develop-generalizationTraining-TrainerController
sankalp04 5 年前
当前提交
121221f2
共有 2 个文件被更改,包括 2 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/learn.py
  2. 1
      ml-agents/mlagents/trainers/trainer_controller.py

1
ml-agents/mlagents/trainers/learn.py


)
trainer_config = load_config(trainer_config_path)
reset_param_dict = load_config(reset_param_dict_path)
env_factory = create_environment_factory(
env_path,
docker_target_name,

1
ml-agents/mlagents/trainers/trainer_controller.py


class TrainerController(object):
# Type of reset_param_dict unspecified as typing library does not support heterogeneous dictionary types
def __init__(
self,
trainers: Dict[str, Trainer],

正在加载...
取消
保存