批数据训练

最新推荐文章于 2023-12-12 14:53:12 发布

Vincent123Mei

最新推荐文章于 2023-12-12 14:53:12 发布

阅读量210

点赞数

分类专栏： pytorch框架文章标签： python 深度学习神经网络

本文链接：https://blog.csdn.net/Vincent123Mei/article/details/107499311

版权

pytorch框架专栏收录该内容

3 篇文章 0 订阅

订阅专栏

import torch
import torch.utils.data as Data  # 处理数据的工具包

# hyper parameter
BATAC_SIZE = 5

x = torch.linspace(1, 10, 10)
y = torch.linspace(10, 1, 10)
# print(x.type()) # torch.FloatTensor

# 先转换成 torch 能识别的 Dataset
torch_dataset = Data.TensorDataset(x, y) # 组装成一个tuple

# 把dataset 放入DataLoader
loader = Data.DataLoader(
    dataset=torch_dataset,
    batch_size=BATAC_SIZE,
    shuffle=True,   # 打乱顺序
    # num_workers=2,  # 多线程
)

for epoch in range(3):
    for step, (batch_x, batch_y) in enumerate(loader):
        print('Epoch: ', epoch, '| Step: ', step, '| batch x: ',
              batch_x.numpy(), '| batch y: ', batch_y.numpy())