PyTorch之Dataset和TensorDataset
Deep Learning系列 @cxxDataset v.s. TensorDataset使用Pytorch搭建过neural network的小伙伴们都知道,在数据准备步骤里,我们需要把训练集的x和y分装在dataset里,然后将dataset分装到DataLoader中去,便于之后在搭建好的模型中训练。简言之,dataset是用来做打包和预处理(比如输入资料路径自动读取);DataLoader则是将整个资料集(dataset)按照batch进行迭代分装或者shuffle(可以得到一个iterat
原创
2021-02-02 22:07:49 ·
9380 阅读 ·
3 评论