pytorch神经网络批数据训练

下面展示批数据训练 代码。

import torch
import torch.utils.data as Data
BATCH_SIZE=5
x=torch.linspace(1, 10, 10)
y=torch.linspace(10, 1, 10)
torch_dataset=Data.TensorDataset(x,y)
loader=Data.DataLoader(
    dataset=torch_dataset ,
    batch_size=BATCH_SIZE ,
    shuffle= True ,  #随机打乱处理的数据
    num_workers=0,  #dataloader一次性创建num_worker个工作进程
)
for epoch in range(3):  #训练全部数据三次
    for step,(batch_x,batch_y) in enumerate(loader):   #enumerate返回(枚举)对象。
        print('Epoch: ', epoch, '| Step: ', step, '| batch x: ',
              batch_x.numpy(), '| batch y: ', batch_y.numpy())

结果