A TFX equivalent for Pytorch



我最近使用TensorFlow Extended(TFX)平台。由于我的开发背景是在Pytorch堆栈上,我正在寻找Pytorch的TFX的良好兼容替代品。

在寻找这个替代方案时,我遇到了torchx和torchserve。但是torchx是一个新开发的库,没有TFX那么全面。另一方面,torchserve的重点是机器学习操作(MLOps)的部署和软件工程方面,并没有像数据管道的TFX那样有很多丰富的功能。

是否有其他Pytorch堆栈的替代方案?

这可能有点自私,因为我是拟议框架的共同创建者之一,但由于问题直接要求建议,我认为在此免责声明是公平的。

如何使用ZenML。它是一个类似于tfx的框架,适用于包括PyTorch在内的所有框架(确实在后台使用ml-pipelines-sdk-尽管它将很快被删除)。

以下是方便您使用的文档和一个pytorch示例,以帮助您入门。这是一个关于将简单的PyTorch代码转换为ZenML管道的最新博客。

干杯,请随意留下评论,这个答案太市场化了,我将删除它(它不是故意的,我今天只是偶然发现了这个问题)。

相关内容

  • 没有找到相关文章

最新更新