pytorch数据分类

Pytorch container

pytorch with cuda and apex

Base image with pytorch and flash attention

基于PyTorch的mnist数据集的分类 这里本人选用LeNet的卷积神经网络结构实现分类，实验训练10个epoch准确率高达99%，测试集准确率达99%。实现代码中对LeNet网络模型进行了一点改动，且模型代码定义在Build_LeNet_for_mnist.py文件中，数据加载不是从网上下载的