Skip to main content

rct 模型的集线器

项目描述

rct-模型池

安装

rctmodelpool您可以通过以下方式安装 git pip

pip install git+https://github.com/rct-ai/rct-modelpool

或者

pip install rctmodelpool

用法

您可以像这样从包中导入rctmodelpool.modelpool

sync_model

from rctmodelpool import modelpool

path = modelpool.sync_model("rcthub://CPM.csv")
print(path)

同步 tgz

import os
import tarfile
from rctmodelpool import modelpool
from pathlib import Path

local_path = modelpool.sync_model("rcthub://"+config.bert_model_path)
parent_path = Path(local_path).parent
print("rcthub sync:",local_path)
modeltar = tarfile.open(local_path)
inside_folder = os.path.commonprefix(modeltar.getnames())
real_model_path = os.path.join(parent_path, inside_folder)
print('model folder:', real_model_path)
if not os.path.exists(real_model_path):
    modeltar.extractall(parent_path)
    print('tar extraction:', parent_path, local_path)

项目详情


下载文件

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

源分布

rctmodelpool-0.1.9.tar.gz (6.7 kB 查看哈希

已上传 source

内置分布

rctmodelpool-0.1.9-py3-none-any.whl (8.4 kB 查看哈希

已上传 py3