
NIN:基于PyTorch的NIN实现及在CIFAR-10数据集上的测试结果
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了使用PyTorch框架实现神经网络在微型图片分类任务中的应用,具体展示了针对CIFAR-10数据集的NIN模型训练过程和测试性能。
使用Pytorch实现NIN
本段落在网络中非官方实施:
0. 开发环境
Docker Image - tensorflow/tensorflow:2.4.0-gpu-jupyter
Library - Pytorch : Stable (1.7.1) - Linux - Python - CUDA (11.0)
使用单GPU
1. 实施细节
- model.py:NIN模型
- train.py:训练NIN
- utils.py:计算正确的预测
- best.pt:最好的NIN权重文件
- NIN-Cifar 10.ipynb:安装库,下载数据集,预处理,训练和结果可视化
- Feature Map.ipynb:可视化完整激活的特征图,前10%的激活细节
辍学的NIN很难训练,并且获得与论文相同的分数。遵循官方法规的详细信息如下:
- 批次大小 128
- 动量 0.9
- 权重衰减 0.00001
全部评论 (0)
还没有任何评论哟~


