当要换成自己的数据集时,需要用Dataset、Dataloader处理数据集,将数据喂给模型
一 基础使用
https://blog.csdn.net/He3he3he/article/details/105441083
https://blog.csdn.net/zw__chen/article/details/82806900
https://blog.csdn.net/u011995719/article/details/85102770
https://blog.csdn.net/xuan_liu123/article/details/101145366
想要得到的数据类型:tensor类型,且shape为:batch_size * input_size
batch_size为3 变量为tensor([...]) 标签为tensor([...])
变量 batch:
tensor[[...],[...],[...]]
标签 batch:
tensor[[...],[...],[...]]
最简单的实现: