【发布时间】:2019-11-06 12:39:12
【问题描述】:
我正在尝试通过 PyTorch 训练分类器。但是,当我向模型提供训练数据时,我遇到了训练问题。
我在y_pred = model(X_trainTensor) 收到此错误:
RuntimeError: 标量类型 Float 的预期对象,但参数 #4 'mat1' 的标量类型 Double
以下是我的代码的关键部分:
# Hyper-parameters
D_in = 47 # there are 47 parameters I investigate
H = 33
D_out = 2 # output should be either 1 or 0
# Format and load the data
y = np.array( df['target'] )
X = np.array( df.drop(columns = ['target'], axis = 1) )
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size = 0.8) # split training/test data
X_trainTensor = torch.from_numpy(X_train) # convert to tensors
y_trainTensor = torch.from_numpy(y_train)
X_testTensor = torch.from_numpy(X_test)
y_testTensor = torch.from_numpy(y_test)
# Define the model
model = torch.nn.Sequential(
torch.nn.Linear(D_in, H),
torch.nn.ReLU(),
torch.nn.Linear(H, D_out),
nn.LogSoftmax(dim = 1)
)
# Define the loss function
loss_fn = torch.nn.NLLLoss()
for i in range(50):
y_pred = model(X_trainTensor)
loss = loss_fn(y_pred, y_trainTensor)
model.zero_grad()
loss.backward()
with torch.no_grad():
for param in model.parameters():
param -= learning_rate * param.grad
【问题讨论】:
-
它是否告诉您触发运行时错误的代码行?
-
是的,在我的最后一个代码块中。
y_pred = model(X_trainTensor)触发它。 -
我不使用 PyTorch,但你可以使用
model(float(X_trainTensor)) -
然后我在同一行收到以下错误:
ValueError: only one element tensors can be converted to Python scalars -
另外,如果我将张量转换为所有浮点数。我收到一个新错误:
AttributeError: 'builtin_function_or_method' object has no attribute 'dim'
标签: python neural-network deep-learning classification pytorch