引例
最近用paddle重写pytorch项目代码时,遇到了DataLoader一直报错的问题,看API文档时,发现两个框架并无区别,于是简单拿来重用,结果调试浪费了很多时间,还是要看API的源代码,比较底层实现。现将问题记录如下:
Paddle读取数据主要用到两个类:paddle.io.Dataset和paddle.io.DataLoader。
下面的例子来自于官方文档:
import numpy as np
from paddle.io import Dataset, DataLoader
import paddle
# define a random dataset
class RandomDataset(Dataset):
def __init__(self, num_samples):
self.num_samples = num_samples

本文通过实例演示了如何使用Paddle中的Dataset和DataLoader,并详细分析了两种常见错误:返回非numpy.ndarray类型数据及返回字典类型数据时所引发的问题。
最低0.47元/天 解锁文章
1153





