实用工具

这包含TorchX实用组件，这些组件是开箱即用的ready-to-use。这些组件简单地执行众所周知的二进制文件（例如cp），并旨在作为教程材料或工作流中具有意义的阶段之间的粘合操作。

torchx.components.utils.echo(msg: str = 'hello world', image: str = 'ghcr.io/pytorch/torchx:0.1.1', num_replicas: int = 1) → torchx.specs.api.AppDef

将消息回显到标准输出（调用 echo）

Parameters

• msg – 要回显的信息

• 图像 – 使用的图像

• num_replicas – 运行的副本数量

torchx.components.utils.touch(file: str, image: str = 'ghcr.io/pytorch/torchx:0.1.1') → torchx.specs.api.AppDef

触碰一个文件（调用 touch）

Parameters

• 文件 – 创建的文件

• 图像 – 要使用的图像

torchx.components.utils.sh(*args: str, image: str = 'ghcr.io/pytorch/torchx:0.1.1', num_replicas: int = 1) → torchx.specs.api.AppDef

通过 sh 运行提供的命令。目前 sh 不支持环境变量替换。

Parameters

• args – 命令行参数

• 图像 – 使用的图像

• num_replicas – 运行的副本数量

torchx.components.utils.copy(src: str, dst: str, image: str = 'ghcr.io/pytorch/torchx:0.1.1') → torchx.specs.api.AppDef

复制文件从源路径到目标路径。源路径和目标路径可以是任何有效的 fsspec URL。

这不支持递归复制或目录。

Parameters

• src – 源 fsspec 文件位置

• dst – 目标fsspec文件位置

• 图像 – 包含复制应用的图像

torchx.components.utils.python(*args: str, m: Optional[str] = None, c: Optional[str] = None, image: str = 'ghcr.io/pytorch/torchx:0.1.1', name: str = 'torchx_utils_python', host: str = 'aws_t3.medium', num_replicas: int = 1) → torchx.specs.api.AppDef

运行 python -c CMD 或 python -m MODULE 在指定的图像和主机上。使用 -- 来分离组件参数和程序参数（例如 torchx run utils.python --m foo.main -- --args to --main）

Parameters

• 参数 – 传递给程序的参数，位于 sys.argv[1:] 中（在使用–c时被忽略）

• m – 以脚本方式运行库模块

• c – 作为字符串传递的程序（如果调度程序对参数长度有限制，则可能会出错）

• 图像 – 要运行的图像

• 名称 – 工作的名称

• 主机 – 一个注册的命名资源

• num_replicas – 运行的副本数量（每个副本都在自己的容器中）

Returns

torchx.components.utils.booth(x1: float, x2: float, trial_idx: int = 0, tracker_base: str = '/tmp/torchx-util-booth', image: str = 'ghcr.io/pytorch/torchx:0.1.1') → torchx.specs.api.AppDef

评估 booth 函数，f(x1, x2) = (x1 + 2*x2 - 7)^2 + (2*x1 + x2 - 5)^2。 输出结果可通过 FsspecResultTracker(outdir)[trial_idx] 访问

Parameters

• x1 – x1

• x2 – x2

• trial_idx – 如果不运行超参数优化，请忽略

• tracker_base – 跟踪器基础输出目录的URI（例如 s3://foo/bar）

• 图像 – 包含展位应用程序的图像