反馈

MMLBaseEstimatorRunConfig 类

  • 参考

所有估算器运行配置的抽象基类。

已弃用。 使用 RunConfiguration 类。

继承
RunConfiguration
MMLBaseEstimatorRunConfig

构造函数

MMLBaseEstimatorRunConfig(compute_target, vm_size=None, vm_priority=None, entry_script=None, script_params=None, node_count=None, process_count_per_node=None, distributed_backend=None, use_gpu=None, use_docker=None, custom_docker_base_image=None, custom_docker_image=None, image_registry_details=None, user_managed=False, conda_packages=None, pip_packages=None, environment_definition=None, inputs=None, source_directory_data_store=None, shm_size=None)

参数

compute_target
AbstractComputeTarget 或 str
必需

发生训练的计算目标。 可以是对象或字符串“local”。

vm_size
str
默认值: None

将为训练创建的计算目标的 VM 大小。

支持的值：任何 Azure VM 大小。

vm_priority
str
默认值: None

将为训练创建的计算目标的 VM 优先级。 如果未指定，则使用“dedicated”。

支持的值：“dedicated”和“lowpriority”。

这仅在输入中指定了 vm_size 参数时才会生效。

entry_script
str
默认值: None

用于开始训练文件的相对路径。

script_params
dict
默认值: None

包含作为自变量传递给 entry_script 的参数的字典。

node_count
int
默认值: None

计算目标中用于训练的节点数。 分布式训练（node_count> 1）仅支持 AmlCompute 目标。

process_count_per_node
int
默认值: None

将 MPI 用作执行后端时，表示每个节点的进程数。

distributed_backend
str
默认值: None

分布式训练的通信后端。

支持的值：“mpi”和“ps”。

'mpi'：MPI/Horovod 'ps'：参数服务器

在 node_count、process_count_per_node、worker_count 或 parameter_server_count> 1 时，此参数是必需的。

当 node_count == 1 且 process_count_per_node == 1 时，除非显式设置后端，否则不会使用后端。 分布式训练仅支持 azureml.core.compute.AmlCompute 目标。

use_gpu
bool
默认值: None

指定要运行试验的环境是否应支持 GPU。 如果为 true，则将在环境中使用基于 GPU 的默认 Docker 映像。 如果为 false，则使用基于 CPU 的映像。 仅当未设置 custom_docker_image 参数时，才会使用默认 docker 映像（CPU 或 GPU）。 此设置仅用于启用了 Docker 的计算目标。

use_docker
bool
默认值: None

指定要运行试验的环境是否应基于 Docker。

custom_docker_base_image
str
默认值: None

要从中生成用于训练的映像的 Docker 映像的名称。

已弃用。 使用 custom_docker_image 参数。

如果未设置，则将使用基于 CPU 的默认映像作为基础映像。

custom_docker_image
str
默认值: None

要从中生成用于训练的映像的 Docker 映像的名称。 如果未设置，则将使用基于 CPU 的默认映像作为基础映像。

image_registry_details
ContainerRegistry
默认值: None

Docker 映像注册表的详细信息。

user_managed
bool
默认值: False

指定 Azure ML 是否重复使用现有的 Python 环境。 如果为 false，将基于 conda 依赖项规范创建 Python 环境。

conda_packages
list
默认值: None

表示要添加到此试验的 Python 环境中的 conda 包的字符串列表。

pip_packages
list
默认值: None

表示要添加到此试验的 Python 环境中的 pip 包的字符串列表。

environment_definition
Environment
默认值: None

试验的环境定义。 它包括 PythonSection、DockerSection 和环境变量。 使用此参数可以设置不通过其他参数直接公开到估算器构造的任何环境选项。 如果指定了此参数，它将优先于其他与环境相关的参数，如 use_gpu、custom_docker_image、conda_packages 或 pip_packages。对于这些无效组合，将报告错误。

inputs
list
默认值: None

要用作输入的 DataReference 或 DatasetConsumptionConfig 对象的列表。

source_directory_data_store
str
默认值: None

项目共享的支持数据存储。

shm_size
str
默认值: None

Docker 容器的共享内存块的大小。 有关详细信息，请参阅 Docker 运行参考。 如果未设置，则使用默认的 azureml.core.environment._DEFAULT_SHM_SIZE。