实用工具¶
这包含TorchX实用组件，这些组件是开箱即用的ready-to-use。这些组件简单地执行众所周知的二进制文件（例如cp），并旨在作为教程材料或工作流中具有意义的阶段之间的粘合操作。
- torchx.components.utils.echo(msg: str = 'hello world', image: str = 'ghcr.io/pytorch/torchx:0.1.1dev0', num_replicas: int = 1) → torchx.specs.api.AppDef[source]¶
将消息回显到标准输出（调用 echo）
- Parameters
msg – 要回显的信息
图像 – 使用的图像
num_replicas – 运行的副本数量
- torchx.components.utils.touch(file: str, image: str = 'ghcr.io/pytorch/torchx:0.1.1dev0') → torchx.specs.api.AppDef[source]¶
触碰一个文件（调用 touch）
- Parameters
文件 – 创建的文件
图像 – 要使用的图像
- torchx.components.utils.sh(*args: str, image: str = 'ghcr.io/pytorch/torchx:0.1.1dev0', num_replicas: int = 1) → torchx.specs.api.AppDef[source]¶
通过 sh 运行提供的命令。目前 sh 不支持环境变量替换。
- Parameters
args – 命令行参数
图像 – 使用的图像
num_replicas – 运行的副本数量
- torchx.components.utils.copy(src: str, dst: str, image: str = 'ghcr.io/pytorch/torchx:0.1.1dev0') → torchx.specs.api.AppDef[source]¶
复制文件从源路径到目标路径。源路径和目标路径可以是任何有效的 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.1dev0', name: str = 'torchx_utils_python', host: str = 'aws_t3.medium', num_replicas: int = 1) → torchx.specs.api.AppDef[source]¶
运行
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.1dev0') → torchx.specs.api.AppDef[source]¶
评估 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）
图像 – 包含展位应用程序的图像