文件名称:LeNet-5:LeNet-5的PyTorch实施和实时可视化
文件大小:3KB
文件格式:ZIP
更新时间:2024-05-24 12:27:06
machine-learning deep-neural-networks deep-learning cnn pytorch
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月。
【文件预览】:
LeNet-5-master
----run.py(3KB)
----requirements.txt(86B)
----README.md(967B)
----lenet.py(2KB)
----.gitignore(33B)