train_loader = data_utils.DataLoader(datase=train_tensor, batch_size=batch_size, shuffle=True)
时间: 2023-06-10 11:03:17 浏览: 116
python torch.utils.data.DataLoader使用方法
这是一个数据加载器,用于将训练数据划分为批次进行训练。其中,train_tensor是训练数据集,batch_size是每个批次的大小,shuffle=True则表示在每次迭代时都打乱数据顺序,以增强训练效果。
阅读全文