ValueError:具有IterableDataset的DataLoader:应为未指定的shuffle选项,但得到shuffle=Tru

2024-09-28 21:02:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用Databunch.create创建使用以下代码行:

ds_train = PyTorchDataset('fld/training/', batch_size = 8)
ds_val = PyTorchDataset('fld/alidation/', batch_size = 8)
from torch.nn.modules.loss import CrossEntropyLoss
model = BaseCNN(n_channels = 1, n_classes = 2)
loss_func = CrossEntropyLoss(weight = torch.Tensor([0.1,0.9]))
data = DataBunch.create(ds_train,ds_val,collate_fn=cnn_collate,path="fld/")

create函数引发了以下错误:

ValueError:具有IterableDataset的DataLoader:应为未指定的shuffle选项,但得到shuffle=True

我不知道我错过了什么。谁能帮我一下吗。 提前谢谢。你知道吗


Tags: 代码sizecreatebatchdstrainvaltorch