pytorch分类

https://github.com/pytorch/serve

CUDA + PyTorch

本文介绍如何使用pytorch搭建基础的神经网络，解决多分类问题。主要介绍了两个模型：①全连接层网络；②VGG11卷积神经网络模型(下次介绍)。为了演示方便，使用了Fashion-Mnist服装分类数据集（10分类数据集，介绍可以去网上搜一下，这里不赘述），也可以在自己的制作的数据集上训练（后面会稍

torch.utils,data.dataset => 用于下载和预处理数据集 torch.utils.data.dataloader => 用来加载数据集并且声称一个个的batch，打乱或者顺序排列。 torch.nn 层的结构，损失函数，都来自于这类，构建模型都是继基类nn.Module。 to