该软件包使用源自 Trax 的想法为 Flax 提供了额外的灵活性
项目描述
亚麻额外
该软件包使用源自Trax的想法为Flax提供了额外的灵活性。
额外内容包括：
- 支持由Redex支持的组合器（串行、并行、分支等） 。
- 一个模块化的训练循环，可以很好地与Optax优化器和指标配合使用。
- 可插入的日志记录到标准输出。
- 并行执行训练和评估任务。
- 常规和最佳指标检查点。
- 张量板集成。
- 额外的亚麻模块。
查看文档、介绍性教程或示例，其中包括：
- 感知器 IO
- 对来自ImageNet的图像进行预训练的分类模型。
- 使用通过结合英语维基百科和 C4获得的大型文本语料库预训练的掩码语言模型。
- 在Kinetics-700-2020数据集的多模式输入（音频、视频和标签）上预训练的自动编码器模型。
下载文件
下载适用于您平台的文件。如果您不确定要选择哪个，请了解有关安装包的更多信息。
源分布
flax-extra-0.0.2.tar.gz
(39.8 kB
查看哈希)
内置分布
flax_extra-0.0.2-py3-none-any.whl
(61.6 kB
查看哈希)
关
flax_extra -0.0.2-py3-none-any.whl 的哈希值
| 算法 | 哈希摘要 | |
|---|---|---|
| SHA256 | aa793d3364024cf38ba158212e6ef96ae4a76633030552ca463a0030b463ef19 |
|
| MD5 | 9364a7fe1a0a05bc4d19c0b3c26279be |
|
| 布莱克2-256 | 76102f3b7195bd6b2c4084ad53a689a8425b78cf64cccde4f1df8dfa38221d17 |