今天小编就为大家分享一篇Pytorch 神经网络—自定义数据集上实现教程，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

第一步、导入需要的包 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, u