我正在尝试创建一个自定义pytorch数据集以插入到DataLoader中,该数据集由单通道图像(20000 x 1 x 28 x 28)、单通道蒙版(20000 x 1 x 28 x 28)和三个标签中。(train_data, batch_size=128, shuffle=True) 我希望trainloader中的单个批次的形状是([128, 1, 28, 28], [128, 1, 28, 28]),无论是左边的图像还是右边的</em
最初,数据加载器是用某些示例创建的。在训练过程中,我需要更换一个数据采集器中的样本。如何把它替换成数据采集器。train_dataloader = DataLoader(train_data, sampler=train_sampler, batch_size=batch_size)
prediction of model
select misclassifi