Pytorch 如何将torch自定义数据集与fastai数据加载器一起使用

我创建了一个定制的火炬数据集，并指定了两个必需的方法：

\uuu getitem\uuuuuu

和

\uuuuu len\uuuuu

。然后我创建了两个火炬数据加载器：

train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=32, shuffle=True, num_workers=2)
val_loader = torch.utils.data.DataLoader(val_dataset, batch_size=32, shuffle=True, num_workers=2)

我想将它们用于

fastai

，因此我尝试运行：

from fastai.vision.data import DataLoaders
dls = vision.data.DataLoaders(train_loader, valid_loader)
from fastai.vision.learner import cnn_learner, error_rate
learner_original = cnn_learner(dls, models.resnet34, metrics=error_rate, pretrained=True)

但是，这给了我一个错误：

AttributeError:“DataLoader”对象没有“after\u batch”属性

我的设置有什么问题？torch数据集是否应该具有另一个属性以使其与fastai数据加载器兼容