跳转至

欢迎

GitHub PyPI GitHub commit activity PyPI - Status

Geyser 是一个低代码任务编排与运行框架,适用于小任务之间的解耦与参数化定义。

特点:

  • 能够支持多种配置文件格式,包括 JSON、YAML、TOML、HOCON 以及 Apple pList。
  • 有向无环图( DAG )并行化任务,提高任务执行效率
  • 低代码运行,任务参数与代码解耦

快速使用

首先,安装geysergeyser-lava

pip install geyser geyser-lava

执行:

geyser -d hello/hello.yaml

安装成功能够输出以下信息:

(2021-09-30 09:52:51,010)[DEBUG][2328932][140688370164544][geyser_lava.task.hello.Hello]: Hello, level debug.
(2021-09-30 09:52:51,010)[INFO][2328932][140688370164544][geyser_lava.task.hello.Hello]: Hello, level info.
(2021-09-30 09:52:51,010)[WARNING][2328932][140688370164544][geyser_lava.task.hello.Hello]: Hello, level warning
(2021-09-30 09:52:51,010)[ERROR][2328932][140688370164544][geyser_lava.task.hello.Hello]: Hello, level error.
(2021-09-30 09:52:51,010)[CRITICAL][2328932][140688370164544][geyser_lava.task.hello.Hello]: Hello, level critical.

核心安装

如果你希望完全自己实现所有任务,可以只安装geyser,即:

pip install geyser

扩展安装

支持 PyTorch

依照PyTorch的安装方式,安装pytorchtorchvisiontorchaudio,并依照PyTorch Ignite的安装方式,安装pytorch-ignite

验证安装,在 CIFAR-10 数据集训练 ResNet-50 模型:

geyser torch/train_resnet_cifar10.yaml