跳转到主要内容

Orbax Checkpoint

项目描述

Orbax Checkpointing

pip install orbax-checkpoint (最新PyPi版本) OR

pip install 'git+https://github.com/google/orbax/#subdirectory=checkpoint' (从此仓库,在HEAD)

import orbax.checkpoint

Orbax包括一个针对JAX用户的检查点库,支持不同框架所需的各种不同功能,包括异步检查点、各种类型和各种存储格式。我们的目标是提供一个高度可定制和可组合的API,以最大限度地提高各种用例的灵活性。

项目详情


下载文件

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

源代码分发

orbax_checkpoint-0.6.4.tar.gz (194.8 kB 查看哈希值)

上传时间 源代码

构建分发

orbax_checkpoint-0.6.4-py3-none-any.whl (270.5 kB 查看哈希值)

上传时间 Python 3

支持者

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF 赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误日志 StatusPage StatusPage 状态页面