如何在pytorch中运行一批？

Question

我是 AI 的新手，python 我正在尝试运行只有一批旨在 overfit.I 找到代码： iter(train_loader).next()

但我不确定在我的代码中在哪里实现它。即使我这样做了，我如何在每次迭代后检查以确保我训练的是同一批次？

train_loader = torch.utils.data.DataLoader(
    dataset_train,
    batch_size=48,
    shuffle=True,
    num_workers=2
)

net = nn.Sequential(
    nn.Flatten(),
    nn.Linear(128*128*3,10)
)


nepochs = 3
statsrec = np.zeros((3,nepochs))

loss_fn = nn.CrossEntropyLoss()
optimizer = optim.Adam(net.parameters(), lr=0.001)


for epoch in range(nepochs):  # loop over the dataset multiple times

    running_loss = 0.0
    n = 0
    for i, data in enumerate(train_loader, 0):
        inputs, labels = data
        
         # Zero the parameter gradients
        optimizer.zero_grad() 

        # Forward, backward, and update parameters
        outputs = net(inputs)
        loss = loss_fn(outputs, labels)
        loss.backward()
        optimizer.step()
    
        # accumulate loss
        running_loss += loss.item()
        n += 1
    
    ltrn = running_loss/n
    ltst, atst = stats(train_loader, net)
    statsrec[:,epoch] = (ltrn, ltst, atst)
    print(f"epoch: {epoch} training loss: {ltrn: .3f}  test loss: {ltst: .3f} test accuracy: {atst: .1%}")

请给我提示

Answer 1

如果您希望对单个批次进行训练，请移除数据加载器上的循环：

for i, data in enumerate(train_loader, 0):
    inputs, labels = data

并且在循环历元之前简单地获取train_loader迭代器的第一个元素，否则next将在每次迭代时被调用，你将运行每个时期在不同的批次上：

inputs, labels = next(iter(train_loader)) i = 0 for epoch in range(nepochs): optimizer.zero_grad() outputs = net(inputs) loss = loss_fn(outputs, labels) loss.backward() optimizer.step() # ...

如何在pytorch中运行一批？

How to run one batch in pytorch?

python

deep-learning

pytorch

artificial-intelligence