batch由1改成2报错

发布网友发布时间：2023-08-01 07:33

共1个回答

热心网友时间：2024-11-16 05:36

原因是因为输入dataloader的图片大小不一致。
提供了两个数据原语：分别是 torch.utils.data.Dataset和torch.utils.data.DataLoader可以在预加载的数据集或者自己的数据集上使用。
其中Dataset表示存储样本及其对应的标签，用于表示数据集的抽象类。
DataLoader包裹一个可迭代的迭代器，这使得Dataset便于访问样品。包装数据集并提供对基础数据的访问。

声明：本网页内容为用户发布，旨在传播知识，不代表本网认同其观点，若有侵权等问题请及时与本网联系，我们将在第一时间删除处理。E-MAIL:11247931@qq.com