摘要: Dataset 是 PyTorch 中用来表示数据集的一个抽象类,我们的数据集可以用这个类来表示,至少需要覆写下面两个方法: 1)__len__:一般用来返回数据集大小。 2)__getitem__:实现这个方法后,可以通过下标的方式 dataset[i] 的来取得第 $i$ 个数据。 DataLo 阅读全文
posted @ 2020-12-02 16:46 _yanghh 阅读(4521) 评论(0) 推荐(1) 编辑