如何使用Pytorch在自定义数据集上构建神经网络？

本文共计891个文字，预计阅读时间需要4分钟。

第一步、导入所需包：pythonimport osimport scipy.io as sioimport numpy as npimport torchimport torch.nn as nnimport torch.backends.cudnn as cudnnimport torch.optim as optimfrom torch.utils.data import Dataset, DataLoaderfrom torchvision import __name__

第一步、导入需要的包

import os import scipy.io as sio import numpy as np import torch import torch.nn as nn import torch.backends.cudnn as cudnn import torch.optim as optim from torch.utils.data import Dataset, DataLoader from torchvision import transforms, utils from torch.autograd import Variable

batchSize = 128 # batchsize的大小 niter = 10 # epoch的最大值

第二步、构建神经网络

设神经网络为如上图所示，输入层4个神经元，两层隐含层各4个神经元，输出层一个神经。