pytorch分类器

PyTorch Lightning image

一、概述 本文在pytorch的框架下，基于softmax分类器的原理，给出softmax分类器的源码，实现图像的分类。结合代码阐述神经网络的搭建过程，用测试集检验神经网络的训练结果，最后在本文末给出全部代码。 二、softmax分类器的搭建 1、下载数据集 import numpy as npim