m0_57909060的博客

原创 Alex模型的构建

1. 前言 在这篇文章中，我将以代码作为主要叙述对象，简要介绍近段时间学习搭建的AlexNet网络和训练工作，更多了解请关注公众号深度之眼，如有侵权请告知删文，具体数据集以及代码我会放到评论区的链接。 2.模型的搭建 2.1 用到的数据库 import os import numpy as np import torch.nn as nn import torch import sys from torch.utils.data import DataLoader import torchvision.tr