4 Star 7 Fork 68

陈雨澄 / research_TF

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

GTN自动迁移工具说明:

该工具读取网络提交者提交的GPU网络模型,config配置文件,用于自动迁移生产NPU网络。

该工具当前支持:

tensorflow1.X,Keras,pytorch框架网络工具自动迁移,

tensorflow(1.X,Keras,2.X),pytorch网络自动添加license。

当前版本:

GTN_V2021.7.1

使用方法:

GPU归档明细:

1、目录名:

网络名_IDxxxx_for_TensorFlow网络名_IDxxxx_for_TensorFlow2.X网络名_IDxxxx_for_PyTorch

2、目录结构:

网络根目录下需包含README.mdrequirements.txt、test目录

test目录下需包含train_full_1p.shtrain_performance_1p.sh

3、config.csv配置:

userid:提交者工号

network:网络名

frame:网络框架,TF1.X,Keras,TF2.X

trainpy_path:keras框架下网络训练脚本入口文件

配置完成后,用户需执行bash gpu_to_npu.sh

目录结构:

----GPU_Input GPU交付件归档目录

----GPU_ALL 迁移成功后GPU代码目录

----NPU_Output 自动迁移临时输出目录

----NPU_PASS 工具自动迁移成功目录

----NPU_FAIL 工具自动迁移失败目录,需网络提交者定位or手动迁移

----log 工具迁移日志,license日志等

----LicenseTool 自动添加license工具

----convert.py 自动迁移,添加license脚本

----config.csv 配置文件,需网络提交者填写

----GPU_Check_InBlue.py 交付件检查工具

----msFmkTransplt pytorch自动迁移工具

空文件

简介

暂无描述 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
Python
1
https://gitee.com/chen-yucheng113/research_TF.git
git@gitee.com:chen-yucheng113/research_TF.git
chen-yucheng113
research_TF
research_TF
master

搜索帮助