我的博客

这个错误通常是因为模型在GPU上训练，但是在CPU上加载模型时出现的。这样就可以在CPU上加载模型了。

import torchimport torch.nn as nnimport torch.optim as optimimport torchvisionimport torchvision.datasets as datasetsimport torchvision.transforms as transformsfrom matplotlib import pyplot as pltfrom torch.utils.data import DataLoader# 超参数定义EPO

减小计算量，内存变少增加feature map，提取更多特征增大感受野融合不同级的特征，得到更准确的效果

一、代码import torchfrom torch import nnimport torchvision.datasets as datasetsimport torchvision.transforms as transformsimport matplotlib.pyplot as pltfrom torch.utils.data import DataLoader#%matplotlib inline# 超参数定义EPOCH = 1BATCH_SIZE = 64TIME_S

一、欠拟合解决方法：1、增加模型参数量（增加问题描述能力）2、增加非线性功能二、过拟合解决方法：1、数据集增广2、正则化3、early stop（早停）4、dropout5、Inception结构