DL Project Template
,简化加载数据、构建网络、训练模型和预测样本的流程。
By C. L. Wang @ 云事业部
工程:
使用方式
下载工程
git clone https://github.com/SpikeKing/DL-Project-Template
创建和激活虚拟环境
virtualenv venv
source venv/bin/activate
安装Python依赖库
pip install -r requirements.txt
开发流程
定义自己的数据加载类,继承DataLoaderBase;
定义自己的网络结构类,继承ModelBase;
定义自己的模型训练类,继承TrainerBase;
定义自己的样本预测类,继承InferBase;
定义自己的配置文件,写入实验的相关参数;
执行训练模型和预测样本操作。
示例工程
识别库中手写数字,工程simp
2021-11-02 14:57:22
138KB
Python
1