1.torch中的Dataset
这个模块更偏重于自己独立的去建立一个新的训练数据集,需要自己去设定参数。
2.torchvision中的datasets
从名字中就可以看到,这个datasets不仅是小写,说明它作用的范围小,而且加了个s,这就说明它是复数,为什么是复数呢,是因为如果我们需要不自己设置训练数据集而去使用官方给的数据集的时候,它里边有。
什么时候用哪个?
大多数情况下两者其实没什么区别。但是如果需要自己设置数据集,需要参数的时候用torch中的Dataset;如果不需要什么参数,或者使用官方数据集的时候使用datasets。