pytorch用自己的数据集进行Dataloader,并对其划分数据集
最近在学pytorch,今天晚上用pytorch的数据加载部分,一开始一直在纠结怎么划分数据集,后来还是手动分了,开始是用torch.utils.data.random_split但是后来一直报错,我也不知道哪里有错,解决不了,后来暴力解决了1.重写dataset类,这是必须要写的主要继承Dataset类,重写__getitem__,and __len__的方法我的问题:针对一个文件夹有n...