我正在使用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
我不知道我错过了什么。谁能帮我一下吗。 提前谢谢。你知道吗
目前没有回答
相关问题 更多 >
编程相关推荐