平台到版本模型、数据、参数以及 git 版本化源代码
项目描述
软件2.0的版本控制系统。
完整文档：stockroom.page
介绍
Stockroom 是一个与 git 版本化源代码一起版本化模型、数据、参数、实验工件等的平台。
- 这很容易。API 与 python 中的字典非常相似
- 它与 Git 一起工作——以防您还需要对源代码进行版本控制。如果你不这样做也没关系。
- 高性能，感谢惊人的机库库
- 与 PyTorch及其生态系统集成，因此您无需编写复杂的管道代码。
为什么
库房最初设计背后的一个重要动机是避免用户学习另一种版本控制工具。我们尝试使 API 尽可能地简洁和熟悉。与其他版本控制工具类似，stockroom 让“git”执行checkout
并依赖“git”在分支/提交之间移动。但与其他工具不同的是，我们通过智能 API 引导您的数据访问，因此当您在提交之间遍历时，我们不需要移动巨大的数据文件。
安装
$ pip install stockroom
例子
from stockroom import StockRoom
stock = StockRoom(write=True)
model.load_state_dict(stock.model['resnet50'])
for e in range(epochs):
for i in range(limit):
optimizer.zero_grad()
x, y = stock.data['dataset_name', i]
out = model(x)
loss = criterion(out, y)
loss.backward()
optimizer.step()
if loss < previous_loss:
stock.experiment['loss'] = loss.item()
stock.model['resnet50'] = model.state_dict()
stock.commit('adding a better model')
贡献
我们很乐意将您列入贡献者名单。在提交 PR 之前，请查看贡献者指南。这是我们最新的#Hall-Of-Fame
执照
该项目根据Apache Software License 2.0的条款获得许可
下载文件
下载适用于您平台的文件。如果您不确定要选择哪个，请了解有关安装包的更多信息。
源分布
stockroom-0.2.2.tar.gz
(119.1 kB
查看哈希)
内置分布
stockroom-0.2.2-py3-none-any.whl
(17.1 kB
查看哈希)
关
stockroom-0.2.2- py3 -none-any.whl 的哈希值
| 算法 | 哈希摘要 | |
|---|---|---|
| SHA256 | 169af06a9e51e6305306557ca49bd87cc95ede24ba202dd62f6c85bc680bdcd9 |
|
| MD5 | 4d09a37ed83c363edb41d0fb66afc56c |
|
| 布莱克2-256 | 648797ad80ea0189f3fe3f5fcd86de3e7230c8d5a0b372ecd91b5245faa7401b |