python训练模型需要多少个ecopy_如何用numpy数据和批量大小训练pytorch模型？

我正在学习Pythorch的基础知识，并打算创建一个简单的4层网络来训练虹膜数据集进行分类。在参考了许多教程之后，我写了这段代码。在import pandas as pd

from sklearn.datasets import load_iris

import torch

from torch.autograd import Variable

epochs=300

batch_size=20

lr=0.01

#loading data as numpy array

data = load_iris()

X=data.data

y=pd.get_dummies(data.target).values

#convert to tensor

X= Variable(torch.from_numpy(X), requires_grad=False)

y=Variable(torch.from_numpy(y), requires_grad=False)

print(X.size(),y.size())

#neural net model

model = torch.nn.Sequential(

torch.nn.Linear(4, 10),

torch.nn.ReLU(),

torch.nn.Dropout(),

torch.nn.Linear(10, 5),

torch.nn.ReLU(),

torch.nn.Dropout(),

torch.nn.Linear(5, 3),

torch.nn.Softmax()

)

print(model)

# Loss and Optimizer

optimizer = torch.optim.Adam(model.parameters(), lr=lr)

loss_func = torch.nn.CrossEntropyLoss()

for i in range(epochs):

# Forward pass

y_pred = model(X)

# Compute and print loss.

loss = loss_func(y_pred, y)

print(i, loss.data[0])

# Before the backward pass, use the optimizer object to zero all of the

# gradients for the variables it will update (which are the learnable weights

# of the model)

optimizer.zero_grad()

# Backward pass

loss.backward()

# Calling the step function on an Optimizer makes an update to its parameters

optimizer.step()

目前我面临两个问题。在我想将批大小设置为20。我该怎么做？在

在这个步骤y_pred = model(X)它显示了这个错误

错误

^{pr2}$