跳转到主要内容

JAX中机器学习训练循环的库集合。

项目描述

CLU - Common Loop Utils

此存储库包含编写ML训练循环的常用功能。目标是使训练循环简短且易于阅读(但将常用任务移动到小型库中),同时不删除研究所需的灵活性。

要开始使用,请查看此Colab

https://colab.research.google.com/github/google/CommonLoopUtils/blob/main/clu_synopsis.ipynb

如果您正在寻找使用示例,请参阅

https://github.com/google/flax/tree/main/examples

您还可以在Flax GitHub讨论页面上找到有关CLU的常见问题的答案

https://github.com/google/flax/discussions

注意:目前我们不接受贡献。如果您想为您的用例扩展库,请fork存储库。

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解更多关于 安装包 的信息。

源代码发行版

clu-0.0.12.tar.gz (72.7 kB 查看哈希值)

上传时间 源代码

构建发行版

clu-0.0.12-py3-none-any.whl (101.8 kB 查看哈希值)

上传时间 Python 3