Pytroch 自写训练模板适合入门版包含十五种经典的自己复现的一维模型 1D CNN

训练模板

在毕业之前，决定整理一下手头的代码，自己做1D-CNN这吗久，打算开源一下自己使用的1D-CNN的代码，包括用随机数生成一个模拟的数据集，到自己写的一个比较好的适合入门的基础训练模板，以及自己复现的所有1D-CNN经典模型进行开源，代码已经上传到了GitHub上，接下来我逐个文件进行讲解。由于写的过于详细导致，写完了之后发现最后写了1万9000多字，都超过我本科论文字数了。如果有问题或者有什么写的不完善的地方欢迎里留言交流。
https://github.com/StChenHaoGitHub/1D_Pytorch_Train_demo.git
LeNet-AlexNet-ZFNet: LeNet-AlexNet-ZFNet一二维复现pytorch
VGG: VGG一二维复现pytorch
GoogLeNet: GoogLeNet一二维复现pytorch
ResNet: ResNet残差网络一二维复现pytorch-含残差块复现思路分析
DenseNet: DenseNet一二维复现pytorch
Squeeze: SqueezeNet一二维复现pytorch

在这里插入图片描述

文章目录

- 训练模板
- 1.生成模拟数据集-Create_dataset.py
- 2.数据集打包-Package_dataset.py
- 3.训练代码分解讲解
- - 3.1导入需要的工具包
  - 3.2 读取数据
  - 3.3 划分训练集和测试集
  - 3.4 数据库加载类的实现
  - 3.5 构建dataloader
  - 3.6 选择训练设备
  - 3.7 选择训练模型
  - 3.8 损失函数的选择
  - 3.9 优化器的选择
  - 3.10 初始化保存训练集准确率和测试集准确率的列表
  - 3.11 训练函数详解（关键）
  - 3.12 测试函数解释
  - 3.13 进行训练
  - 3.14 绘制训练测试准确率曲线
- 4.训练完整代码
- 经验总结
- 结束

1.生成模拟数据集-Create_dataset.py

首先在没有数据集的情况下，我们通过最简单的方式来生成一个数据集代码，数据集包含两部分，包括训练的值和标签，训练的值用data表示，训练的标签用label表示

Create_dataset.py其包含代码如下

import numpy as np
# 模拟的样本数量
numbers = 100
# 模拟的样本通道数
channels = 3
# 模拟的信号长度
length = 224
# 模拟的类别
classes = 2

# 生成随机数据 生成的数据维度为（数量，通道，长度）
data = np.random.randn(numbers,channels,length)
# 生成标签
label = np.random.randint(0,classes,numbers)

# 将数据和标签保存到Dataset文件夹里
np.save('Dataset/data.npy',data,allow_pickle=True)
np.save('Dataset/label.npy',label,allow_pickle=True)

np.random.randn 示例代码如下

import numpy as np
data = np.random.randn(100,3,244)
print(data.shape)
# (100, 3, 244)

使用np.random.randint生成标签，改代码的作用是生成numbers个包含于[0,classes）的整数

label = np.random.randint(0,classes,numbers)

np.random.randint 示例代码如下

import numpy as np

label = np.random.randint(0, 2, 10)
print(label)
# [0 0 0 1 1 0 1 0 0 0]

运行之后我们在Dataset文件夹下生成了两个npy文件

2.数据集打包-Package_dataset.py

在pytorch的训练之前，我们需要一个函数将数据和标签打包在一起也就是形成一组[值，标签]，的形式，方便之后对数据进行划分和训练，这是一个默认的通用的规范，如果你是新手，你只需要知道这吗做就对了。
Package_dataset.py代码如下

import numpy as np
def package_dataset(data, label):
    dataset = [[i, j] for i, j in zip(data, label)]
    # channel number
    channels = data[0].shape[0]
    # data length
    length = data[0].shape[1]
    # data classes
    classes = len(np.unique(label))
    return dataset, channels, length, classes


if __name__ == '__main__':
    data = np.load('Dataset/data.npy')
    label = np.load('Dataset/label.npy')
    dataset, channels, length, classes = package_dataset(data, label)
    print(channels, length, classes)
    #  3 224 2

其中输入的数据为data和label返回的是dataset,channels,length,classes
在之后的训练模型的初始化过程我们会用到channels通道数,length特征长度,classes类别个数这三个值
下面的代码是打包值和标签的代码

dataset = [[i, j] for i, j in zip(data, label)]

如果要是新手看着会有点抽象这里也是举一个简单的例子，通过zip()和列表生成式，得到了多个[值，标签]构成的dataset

data = [[1, 2, 3],
        [3, 1, 3],
        [1, 2, 3]]

label = [0,1,0]

dataset = [[i, j] for i, j in zip(data, label)]

print(dataset)
#[[[1, 2, 3], 0], [[3, 1, 3], 1], [[1, 2, 3], 0]]

3.训练代码分解讲解

3.1导入需要的工具包

接下来我们一步一步给出训练代码种的各个部分，首先导入一些用到的工具包

import numpy as np
import torch
import matplotlib.pyplot as plt
from torch.utils.data import DataLoader, Dataset,random_split

导入我们自己写的打包数据集的API

from Package_dataset import package_dataset

导入模型,这里的模型可以根据需要自己选择，如果你是新手就知道这部分都是各种模型就完事了

from Models.LeNet import LeNet
from Models.AlexNet import AlexNet
from Models.ZFNet import ZFNet
from Models.VGG19 import VGG19
from Models.GoogLeNet import GoogLeNet
from Models.ResNet50 import ResNet50
from Models.DenseNet import DenseNet
from Models.SqueezeNet import SqueezeNet
from Models.Mnasnet import MnasNetA1
from Models.MobileNetV1 import MobileNetV1
from Models.MobileNetV2 import MobileNetV2
from Models.MobileNetV3 import MobileNetV3_large, MobileNetV3_small
from Models.shuffuleNetV1 import shuffuleNetV1_G3
from Models.shuffuleNetV2 import shuffuleNetV2
from Models.Xception import Xception

3.2 读取数据

加载数据，把值和标签读取进来

data = np.load('Dataset/data.npy')
label = np.load('Dataset/label.npy')

3.3 划分训练集和测试集

设置训练集和测试集的划分比例dataset_partition_rate。0.7就是训练集和测试集7:3划分
训练轮数总轮数epoch_number
和训练多少轮进行一次测试，这个值为show_result_epoch,为什么需要设置这个值，因为有些时候训练的太快了，如果每一轮都打印，先不说效率问题，就是一直打印你也看不清上面的字

dataset_partition_rate = 0.7
epoch_number = 1000
show_result_epoch = 10

使用我们之前介绍的自己写的APIpackage_dataset,输出dataset, channels, length, classes
dataset为打包好的数据，channels为每个值的通道维度的个数length为通道特征长度，classes训练类别

dataset, channels, length, classes = package_dataset(data, label)

划分数据集，划分数据集使用的是torch.utils.data.random_split,其中random_split函数要输入两个参数，一个是dataset也就是待划分的数据集，还有要传进去一个列表lengths,这个里面要传入的是你要划分出的训练集和测试集的样本的数量，我们之前定义的dataset_partition_rate表示的是训练集样本占总样本的比例，所以需要通过int(len(dataset) * dataset_partition_rate)来获得训练集的样本数量，而且为了保证训练集和测试集的样本总数为数据集样本数量，则需要通过数据集的样本总数减去训练集的样本数量test_len = int(len(dataset)) - train_len,这块如果理解不过来可以简单停下来思考一下。

# partition dataset
train_len = int(len(dataset) * dataset_partition_rate)
test_len = int(len(dataset)) - train_len
train_dataset, test_dataset = random_split(dataset=dataset, lengths=[train_len, test_len])

编写数据库加载类，这个东西是干嘛用的，简单的来说这个类的作用是将原始数据（假定为一系列值和标签的配对）转换为PyTorch能够理解和操作的格式。这使得这些数据可以被PyTorch的数据加载器（DataLoader）等工具用于训练机器学习模型，例如在批处理、随机洗牌和并行处理数据等方面。这个东西也可以直接理解为一个固定的API，并不需要详细了解原理，知道是干什么用的即可。

3.4 数据库加载类的实现

# 数据库加载
class Dataset(Dataset):
    def __init__(self, data):
        self.len = len(data)
        self.x_data = torch.from_numpy(np.array(list(map(lambda x: x[0], data)), dtype=np.float32))
        self.y_data = torch.from_numpy(np.array(list(map(lambda x: x[-1], data)))).squeeze().long()

    def __getitem__(self, index):
        return self.x_data[index], self.y_data[index]

    def __len__(self):
        return self.len

加载dataloder,train_dataset和test_dataset是random_split对输入的dataset依照我们设置的比例，划分出的训练集和测试集，其除了与dataset包含的训练样本的数量不一样之外，其每个训练样本的数据格式还是一样的都是[值，标签]
Train_dataset = Dataset(train_dataset)这句话官方或者最准确的表达是通过实例化Dataset，创建了一个针对 train_dataset 数据准备的特定数据集对象。好吧我也觉得有点不是person话。
这里也不需要纠结就理解成需要通过Dataset分别处理一下train_dataset和test_dataset，然后将分别得到的Train_dataset和Test_dataset,然后再把这两个东西扔到DataLoader里，实例化DataLoader里面有常用到的两个需要设置的变量，shuffle设置为Ture就是对输入的训练集或者测试集进行打乱，然后batch_size指的是训练中使用的Mini-Batch策略种batch的大小。

3.5 构建dataloader

Train_dataset = Dataset(train_dataset)
Test_dataset = Dataset(test_dataset)
dataloader = DataLoader(Train_dataset, shuffle=True, batch_size=50)
testloader = DataLoader(Test_dataset, shuffle=True, batch_size=50)

3.6 选择训练设备

选择训练设备，是选择CPU训练还是GPU训练，下面代码的作用是如果cuda可用一般来说也就是你安装的pytorch是GPU版本的那就默认设备是GPU，如果没有GPU，那torch.cuda.is_available()会返回False就会选择CPU，一般来说我们使用自己的笔记本，或者台式机的时候都是只有一个显卡，如果你的设备是服务器，且安装了多个显卡的情况下，这里的cuda:0可以设置成成其他编号的cuda例如cuda:1 cuda:2以此类推

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

3.7 选择训练模型

此时之前我们在Package_dataset种返回channels,length,classes的作用就出来了，在对模型进行实例化的部分，会需要用到这些参数，在如果在模型构建的过程中使用了自适应池化层，一般就不需要传入样本点数，也就是值的长度，因此一般有两种情况，初始化需要输入通道数，类别个数，和值的长度也就在一维种的样本点个数，或者只需要输入通道数，和类别个数。
其中model.to(device)是将模型部署到我们上一步选择的device上

# 模型初始化
# model = LeNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = AlexNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = AlexNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = ZFNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = VGG19(in_channels=channels, classes=classes)
# model = GoogLeNet(in_channels=channels, classes=classes)
# model =ResNet50(in_channels=channels, classes=classes)
# model =DenseNet(in_channels=channels, classes=classes)
# model =SqueezeNet(in_channels=channels, classes=classes)
# model =MobileNetV1(in_channels=channels, classes=classes)
# model =MobileNetV2(in_channels=channels, classes=classes)
# model =MobileNetV3_small(in_channels=channels, classes=classes)
# model =MobileNetV3_large(in_channels=channels, classes=classes)
# model =shuffuleNetV1_G3(in_channels=channels, classes=classes)
# model =Xception(in_channels=channels, classes=classes)
model =shuffuleNetV2(in_channels=channels, classes=classes)
model.to(device)

3.8 损失函数的选择

损失函数的选择,在对模型实例化了之后需要进一步选择一下用于计算模型训练出的结果和实际结果的不一致程度，也就是损失的大小，下面的代码里使用的交叉误熵损失函数，同时这个损失函数也可以认为是一个模型所以也可以把损失函数使用和将模型移动到 GPU上同样的方法将损失函数也移动到GPU上。在多分类任务中，损失函数直接选用交叉误熵就可以了。

criterion = torch.nn.CrossEntropyLoss()
criterion.to(device)

3.9 优化器的选择

优化器选择，由于深度学习算法的核心思想就是反向传播和梯度下降，其中值得使用的就两种，一种Adam一种SGD加动量，其他的不需要尝试。

# 优化器选择
# optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
optimizer = torch.optim.SGD(model.parameters(), lr=0.001, momentum=0.9)

3.10 初始化保存训练集准确率和测试集准确率的列表

初始化两个列表，这两个列表分别用于存储训练集准确率和测试集准确率，最终代码中，每间隔show_result_epoch轮保存一次训练准确率和测试准确率，打印一次训练集和测试集的准确率

train_acc_list = []
test_acc_list = []

3.11 训练函数详解（关键）

接下来就是最难的部分训练函数的部分了，我会逐个非常细致的讲解，将train函数进行额外的拆解,如果你是小白希望你不要被吓退，看懂这个你就超越了绝大多出只会复制粘贴的人了。首先train函数需要传入的一个参数就是epoch，这个参数是当前的轮数，这个参数的作用是用来判断当前的训练轮数是否为show_result_epoch的整数倍，如果话计算该次训练训练集和准确率和测试集的准确率。

def train(epoch):

设置模型为train模式，之后在预测测试集的时候会执行model.eval()函数，两种状态主要会影响Dorpout和BatchNormilze，以Dorpout为例如果在train模式下，Droput每次预测时候丢提的节点是随机的，但是在eval模式下他丢弃哪个节点是固定的。如果想用复现测试集的结果则必须在训练和测试的时候添加模式切换代码，否则会复现不出结果

	model.train()

train_correct正确的训练集样本个数，train_total用来保存的是所有样本总数，

    train_correct = 0
    train_total = 0

模型的训练步骤，也是整个代码最核心的部分，就是这部分代码实现了深度学习训练的功能下面逐行解释

    for data in dataloader:
        train_data_value, train_data_label = data
        train_data_value, train_data_label = train_data_value.to(device), train_data_label.to(device)
        train_data_label_pred = model(train_data_value)
        loss = criterion(train_data_label_pred, train_data_label)
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()

dataloader = DataLoader(Train_dataset, shuffle=True, batch_size=50)

从迭代器种读取打包好的数据假设dataloder的长度为 $n$ ，这个dataloder是这个代码过来的,可以看到里面的batch_size设置的大小为50，假设我们输入进去的Train_dataset长度为70，则最后dataloder可以通过迭代得到一个长度为50的样本和一个长度为20的样本，假设batch_size设置的值为10，则，会返回7个长度为10的样本,假设batch_size值设为100，则返回一个长度为70的他并非v恶霸，总结一下,一共会得到三种情况。

当batch_size的值大于DataLoader()的输入的第一个参数Train_dataset或者Test_dataset的长度的时候，迭代器只迭代一次数据，数据的大小是输入的Train_dataset或者Test_dataset的长度，也就是只起到打乱的一下的作用，但是没有使用mini_batch策略
当batch_size的值小于DataLoader()的输入的第一个参数Train_dataset或者Test_dataset的长度，且输入的Train_dataset或者Test_dataset的长度是batch_size的整数倍的时候，假设输入的Dataloader的dataset长度为 $n$ batch_size大小为 $b$ ,迭代次数 $t = n ∣ b$ ,这个 $∣$ 代表的是整除，也就是可以迭代 $t$ 次每次迭代的数据长度为 $b$
当batch_size的值小于DataLoader()的输入的第一个参数Train_dataset或者Test_dataset的长度，且输入的Train_dataset或者Test_dataset的长度不是batch_size的整数倍的时候，同样假设Dataloader的dataset长度为 $n$ batch_size大小为 $b$ ，则迭代次数是 $t + 1$ 次同样 $t = n ∣ b$ ，前 $t$ 次的迭代出的数据长度为 $b$ ，最后一次的迭代的数据长度为 $n - t \times b$

    for data in dataloader:

对于每次迭代得到的数据进行下面的操作
首先data数据内包含两个值，一个这个batch中的值另一个是这个batch中的标签

        train_data_value, train_data_label = data

之后将得到的值和标签都放到device中

        train_data_value, train_data_label = train_data_value.to(device), train_data_label.to(device)

然后调用模型进行预测train_data_label_pred为预测到的结果，这里要说一下这个结果的数据维度是(len(data),classes),len(data)本地迭代到的data的数据长度，classes是该分类任务的类别数量。

        train_data_label_pred = model(train_data_value)

调用criterion损失函数计算损失，新手注意输入的预测值和真实标签不是预测标签和真实标签

        loss = criterion(train_data_label_pred, train_data_label)

梯度清零，调用optimizer的zero_grad方法，将为每一个可学习参数保存的梯度都清楚到，需要独立拆除这一步的原因据说是保留的目的是有一些任务需要累计反向传播的梯度。

        optimizer.zero_grad()

反向传播梯度，通过反向传播给每一个学习参数都分配一个梯度

        loss.backward()

参数更新，使用梯度与原有的参数作用一下得到这一轮学习之后的参数，作用的方式就是执行以下梯度下降,具体作用的方式和选用的优化器有关。

        optimizer.step()

下面是计算准确率和测试的部分，epoch为当前训练轮次，show_result_epoch为多少轮查看一次训练集和测试集的准确率还有损失，并且把训练集和测试集的损失记录下来绘制一个准确率的变化曲线。

    if epoch % show_result_epoch == 0:

通过torch.max来获得预测的标签，torch.max返回两个值一个是最大的概率probability，一个是最大值的索引predicted，也就是我们认为的标签。

        probability, predicted = torch.max(train_data_label_pred.data, dim=1)

torch.max示例代码如下

import torch

data = torch.Tensor([[0.6, 0.4],
                     [0.3, 0.7]])


probability, predicted = torch.max(data.data, dim=1)

print(probability)
# tensor([0.6000, 0.7000])
print(predicted)
# tensor([0, 1])

将本次训练的样本个数记录到保存训练样本总个数的变量train_total，至于后面跟着的size(0)是表示的是读取Tensor维度第一个值，例如train_data_label_pred维度为(20,2)20为本次的batch大小2,为类别数量，train_data_label_pred.size(0)就是把20取出来。

        train_total += train_data_label_pred.size(0)

记录预测正确样本的个数predicted == train_data_label将预测标签和真实标签对比，这是一个语法首先这两个Tensor的长度得一致，之后会返回一个包含Ture和Flase的数组，对应位置预测的标签相同为Ture预测的标签不同为False，之后在(predicted == train_data_label)加一个.sum()返回的就是预测正确的样本的个数，一个简单的例子如下.item()返回Tensor中的值。

        train_correct += (predicted == train_data_label).sum().item()

import torch

predicted = torch.Tensor([0, 1, 1, 0, 1])
train_data_label = torch.Tensor([0, 0, 1, 0, 1])

print(predicted == train_data_label)
# tensor([ True, False,  True,  True,  True])
print((predicted == train_data_label).sum())
# tensor(4)
print((predicted == train_data_label).sum().item())
# 4

计算训练集准确率并添加到train_acc_list中，train_correct为预测正确的数量,train_total为预测的样本总数，round的作用是作用域浮点数取小数点后多少位。

        train_acc = round(100 * train_correct / train_total, 4)
        train_acc_list.append(train_acc)

打印准确率和损失

        print('=' * 10, epoch // 10, '=' * 10)
        print('loss:', loss.item())
        print(f'Train accuracy:{train_acc}%')

那吗训练函数的完整代码如下，解读完毕

def train(epoch):
	model.train()
    train_correct = 0
    train_total = 0
    for data in dataloader:
        train_data_value, train_data_label = data
        train_data_value, train_data_label = train_data_value.to(device), train_data_label.to(device)
        train_data_label_pred = model(train_data_value)
        loss = criterion(train_data_label_pred, train_data_label)
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
    if epoch % show_result_epoch == 0:
        probability, predicted = torch.max(train_data_label_pred.data, dim=1)
        train_total += train_data_label_pred.size(0)
        train_correct += (predicted == train_data_label).sum().item()
        train_acc = round(100 * train_correct / train_total, 4)
        train_acc_list.append(train_acc)
        print('=' * 10, epoch // 10, '=' * 10)
        print('loss:', loss.item())
        print(f'Train accuracy:{train_acc}%')
        test()

3.12 测试函数解释

同理测试函数除了没有反向传播计算损失和训练过程之外，其他基本一致，其中唯一多的就是在进入调用迭代器读取测试集样本之前多轮一个with torch.no_grad()用于关闭梯度改变相关的功能，这里我刚看的时候也疑惑我没有进行反向传播为什么还需要锁以下梯度，通过查找，得到的答案是即使不使用梯度对参数进行更新如果不把梯度这块给锁上PyTorch默认会跟踪和存储用于自动梯度计算的中间值会影响运算效率，也要避免在一定情况下即使不用优化器更新梯度也可能会造成梯度的改变，因此使用torch.no_grad()是必要的。

def test():
	model.eval()
    test_correct = 0
    test_total = 0
    with torch.no_grad():
        for testdata in testloader:
            test_data_value, test_data_label = testdata
            test_data_value, test_data_label = test_data_value.to(device), test_data_label.to(device)
            test_data_label_pred = model(test_data_value)
            test_probability, test_predicted = torch.max(test_data_label_pred.data, dim=1)
            test_total += test_data_label_pred.size(0)
            test_correct += (test_predicted == test_data_label).sum().item()
    test_acc = round(100 * test_correct / test_total, 3)
    test_acc_list.append(test_acc)
    print(f'Test accuracy:{(test_acc)}%')

3.13 进行训练

为了不让打印出现第0次训练整体变成了(1, epoch_number+1),这块不难但有时候要是懵住了或者就想知道为啥的话，俺就改成range(epoch_number)跑一下看看即可。

for epoch in range(1, epoch_number+1):
    train(epoch)

3.14 绘制训练测试准确率曲线

plt.plot(np.array(range(epoch_number//show_result_epoch)) * show_result_epoch, train_acc_list)
plt.plot(np.array(range(epoch_number//show_result_epoch)) * show_result_epoch, test_acc_list)
plt.legend(['train', 'test'])
plt.title('Result')
plt.xlabel('Epoch')
plt.ylabel('Accuracy')
plt.show()

训练轮数为1000，最终结果如下，由于我们的数据是随机生成的二分类数据集，所以测试集的准确率最终在50%上下移动。
在这里插入图片描述

4.训练完整代码

希望您在看下面的完整代码之前可以先查看上面的内容，之后您看下面的代码再也不会恐惧会有一种通透的感觉。

import numpy as np
import torch
import matplotlib.pyplot as plt
from torch.utils.data import DataLoader, Dataset,random_split
from Package_dataset import package_dataset

from Models.LeNet import LeNet
from Models.AlexNet import AlexNet
from Models.ZFNet import ZFNet
from Models.VGG19 import VGG19
from Models.GoogLeNet import GoogLeNet
from Models.ResNet50 import ResNet50
from Models.DenseNet import DenseNet
from Models.SqueezeNet import SqueezeNet
from Models.Mnasnet import MnasNetA1
from Models.MobileNetV1 import MobileNetV1
from Models.MobileNetV2 import MobileNetV2
from Models.MobileNetV3 import MobileNetV3_large, MobileNetV3_small
from Models.shuffuleNetV1 import shuffuleNetV1_G3
from Models.shuffuleNetV2 import shuffuleNetV2
from Models.Xception import Xception

data = np.load('Dataset/data.npy')
label = np.load('Dataset/label.npy')

dataset_partition_rate = 0.7
epoch_number = 1000
show_result_epoch = 10

dataset, channels, length, classes = package_dataset(data, label)

# partition dataset
train_len = int(len(dataset) * dataset_partition_rate)
test_len = int(len(dataset)) - train_len
train_dataset, test_dataset = random_split(dataset=dataset, lengths=[train_len, test_len])


# 数据库加载
class Dataset(Dataset):
    def __init__(self, data):
        self.len = len(data)
        self.x_data = torch.from_numpy(np.array(list(map(lambda x: x[0], data)), dtype=np.float32))
        self.y_data = torch.from_numpy(np.array(list(map(lambda x: x[-1], data)))).squeeze().long()

    def __getitem__(self, index):
        return self.x_data[index], self.y_data[index]

    def __len__(self):
        return self.len


# 数据库dataloader
Train_dataset = Dataset(train_dataset)
Test_dataset = Dataset(test_dataset)
dataloader = DataLoader(Train_dataset, shuffle=True, batch_size=50)
testloader = DataLoader(Test_dataset, shuffle=True, batch_size=50)
# 训练设备选择GPU还是CPU
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

# 模型初始化
# model = LeNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = AlexNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = AlexNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = ZFNet(in_channels=channels, input_sample_points=length, classes=classes)
# model = VGG19(in_channels=channels, classes=classes)
# model = GoogLeNet(in_channels=channels, classes=classes)
# model =ResNet50(in_channels=channels, classes=classes)
# model =DenseNet(in_channels=channels, classes=classes)
# model =SqueezeNet(in_channels=channels, classes=classes)
# model =MobileNetV1(in_channels=channels, classes=classes)
# model =MobileNetV2(in_channels=channels, classes=classes)
# model =MobileNetV3_small(in_channels=channels, classes=classes)
# model =MobileNetV3_large(in_channels=channels, classes=classes)
# model =shuffuleNetV1_G3(in_channels=channels, classes=classes)
# model =Xception(in_channels=channels, classes=classes)
model =shuffuleNetV2(in_channels=channels, classes=classes)
model.to(device)

# 损失函数选择
criterion = torch.nn.CrossEntropyLoss()
criterion.to(device)
# 优化器选择
# optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
optimizer = torch.optim.SGD(model.parameters(), lr=0.001, momentum=0.9)

train_acc_list = []
test_acc_list = []


# 训练函数
def train(epoch):
	model.train()
    train_correct = 0
    train_total = 0
    for data in dataloader:
        train_data_value, train_data_label = data
        train_data_value, train_data_label = train_data_value.to(device), train_data_label.to(device)
        train_data_label_pred = model(train_data_value)
        loss = criterion(train_data_label_pred, train_data_label)
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
    if epoch % show_result_epoch == 0:
        probability, predicted = torch.max(train_data_label_pred.data, dim=1)
        train_total += train_data_label_pred.size(0)
        train_correct += (predicted == train_data_label).sum().item()
        train_acc = round(100 * train_correct / train_total, 4)
        train_acc_list.append(train_acc)
        print('=' * 10, epoch // 10, '=' * 10)
        print('loss:', loss.item())
        print(f'Train accuracy:{train_acc}%')
        test()


# 测试函数
def test():
	model.eval()
    test_correct = 0
    test_total = 0
    with torch.no_grad():
        for testdata in testloader:
            test_data_value, test_data_label = testdata
            test_data_value, test_data_label = test_data_value.to(device), test_data_label.to(device)
            test_data_label_pred = model(test_data_value)
            test_probability, test_predicted = torch.max(test_data_label_pred.data, dim=1)
            test_total += test_data_label_pred.size(0)
            test_correct += (test_predicted == test_data_label).sum().item()
    test_acc = round(100 * test_correct / test_total, 3)
    test_acc_list.append(test_acc)
    print(f'Test accuracy:{(test_acc)}%')


for epoch in range(1, epoch_number+1):
    train(epoch)

plt.plot(np.array(range(epoch_number//show_result_epoch)) * show_result_epoch, train_acc_list)
plt.plot(np.array(range(epoch_number//show_result_epoch)) * show_result_epoch, test_acc_list)
plt.legend(['train', 'test'])
plt.title('Result')
plt.xlabel('Epoch')
plt.ylabel('Accuracy')
plt.show()