登录社区云，与社区用户共同成长
邀请您加入社区
对数据集处理虽说很方便但在参数选取和其他的细节方面还容易出问题，尤其是最后一个Batch长度不足，会导致输出维度发生问题，若直接舍去，我还想要全部的数据结果使用方法① 创建一个 Dataset 对象② 创建一个 DataLoader 对象③ 循环这个 DataLoader 对象，将xx, xx加载到模型中进行训练train_loader = DataLoader(dataset, batch_si
