weixin_42996407的博客

原创 《PyTorch深度学习实践》 P8 加载数据集

刘二老师的《PyTorch深度学习实践》，第八讲，自己测试的代码如下：因为loss一直不收敛，所以试着把epoch改为1000，但还是不收敛，在0.5和0.8间反复横跳import numpy as npimport torchimport matplotlib.pyplot as pltfrom torch.utils.data import Dataset, DataLoader# Dataset 帮助构建数据集，使数据集可以被索引# Dataset是抽象类，即不能实例化，只能被继