train_loader = DataLoader(train_dataset, batch_size = 3, shuffle = True)
for batch in train_loader: model.train() x,y=batch pred=model(x)
我在dataloader中使用了map迭代，对于每次迭代，我从dataloader接收一个ind，数据，为什么我们要在代码中预测x（ind）
vcudknz31#
dataloader创建一个torchTensor数据集的堆栈，但是这通常取决于你的数据集。它的堆栈大小取决于你的批量大小，在你的例子中是3。
1条答案
按热度按时间vcudknz31#
dataloader创建一个torchTensor数据集的堆栈，但是这通常取决于你的数据集。它的堆栈大小取决于你的批量大小，在你的例子中是3。