Pytorch搭建神经网络的一个简单框架 1.获取神经网络能用的数据(batch) 2.构建网络(具体看自己实现的网络,先不写) 3.建立优化器和训练 torch使用torch.utils.data.DataLoader读取数据,将数据集转换成最终能用的batch,一般搭建网络都需要这个。 import torch.utils.data as data # dataset 假设dataset是原始数据集 # 先要定义一个class继承torch.utils.data.Dataset class Mydataset(data.Dataset): def __init__(self)