上传者: 42110070
|
上传时间: 2022-01-20 00:44:23
|
文件大小: 3KB
|
文件类型: -
LeNet-5
这实现了略微修改的LeNet-5 [LeCun et al。,1998a],并在上达到了约99%的准确度。
设置
使用以下命令安装所有依赖项
$ pip install -r requirements.txt
用法
启动visdom服务器进行可视化
$ python -m visdom.server
开始训练程序
$ python run.py
请参阅时期火车损耗实时图表。
经过训练的模型将作为ONNX导出到lenet.onnx 。 可以使用查看lenet.onnx文件
参考
[ ] Y. LeCun,L。Bottou,Y。Bengio和P. Haffner。 “基于梯度的学习应用于文档识别。” IEEE会议论文集,86(11):2278-2324,1998年11月。