1 Star 2 Fork 1

wwfu/onnx-helper

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
tensor_fp16.py 393 Bytes
一键复制 编辑 原始数据 按行查看 历史
wwfu 提交于 2022-09-23 05:25 . first commit
#!/usr/bin/python
# -*- coding: utf-8 -*-
import sys
import tqdm
from pathlib import Path
import numpy as np
input_dir = Path(sys.argv[1])
output_dir = Path(sys.argv[2])
output_dir.mkdir(parents=True, exist_ok=True)
for bin_path in tqdm.tqdm(input_dir.iterdir()):
data = np.fromfile(bin_path, np.float32).astype(np.float16)
save_path = output_dir / bin_path.name
data.tofile(save_path)
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/wwfu/onnx-helper.git
[email protected]:wwfu/onnx-helper.git
wwfu
onnx-helper
onnx-helper
master

搜索帮助