computer-vision - RuntimeError:/pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:15____ 不支持多目标
问题描述
我遇到了那个错误 RuntimeError: multi-target not supported at /pytorch/aten/src/THCUNN/generic/ClassNLCriterion.cu:15____
我的输入是 340 的二进制向量,目标是 8 的二进制向量,对于 '"
loss = criterion(outputs, stat_batch)
,我得到outputs.shape
= [64,8] 和stat_batch.shape
=[64,8]
这是模型
class MMP(nn.Module):
def __init__(self, M=1):
super(MMP, self).__init__()
# input layer
self.layer1 = nn.Sequential(
nn.Conv1d(340, 256, kernel_size=1, stride=1, padding=0),
nn.ReLU())
self.layer2 = nn.Sequential(
nn.Conv1d(256, 128, kernel_size=1, stride=1, padding=0),
nn.ReLU())
self.layer3 = nn.Sequential(
nn.Conv1d(128, 64, kernel_size=1, stride=1, padding=0),
nn.ReLU())
self.drop1 = nn.Sequential(nn.Dropout())
self.batch1 = nn.BatchNorm1d(128)
# LSTM
self.lstm1=nn.Sequential(nn.LSTM(
input_size=64,
hidden_size=128,
num_layers=2,
bidirectional=True,
batch_first= True))
self.fc1 = nn.Linear(128*2,8)
self.sof = nn.Softmax(dim=-1)
def forward(self, x):
out = self.layer1(x)
out = self.layer2(out)
out = self.layer3(out)
out = self.drop1(out)
out = out.squeeze()
out = out.unsqueeze(0)
#out = out.batch1(out)
out,_ = self.lstm1(out)
print("lstm",out.shape)
out = self.fc1(out)
out =out.squeeze()
#out = out.squeeze()
out = self.sof(out)
return out
#traiin_model
criterion = nn.CrossEntropyLoss()
if CUDA:
criterion = criterion.cuda()
optimizer = optim.SGD(model.parameters(), lr=LEARNING_RATE, momentum=0.9)
for epoch in range(N_EPOCHES):
tot_loss=0
# Training
for i, (seq_batch, stat_batch) in enumerate(training_generator):
# Transfer to GPU
seq_batch, stat_batch = seq_batch.to(device), stat_batch.to(device)
print(i)
print(seq_batch)
print(stat_batch)
optimizer.zero_grad()
# Model computation
seq_batch = seq_batch.unsqueeze(-1)
outputs = model(seq_batch)
if CUDA:
loss = criterion(outputs, stat_batch).float().cuda()
else:
loss = criterion(outputs.view(-1), stat_batch.view(-1))
print(f"Epoch: {epoch},number: {i}, loss:{loss.item()}...\n\n")
tot_loss += loss.item(print(f"Epoch: {epoch},file_number: {i}, loss:{loss.item()}...\n\n"))
loss.backward()
optimizer.step()
解决方案
您的目标stat_batch
必须具有(64,)
因为nn.CrossEntropyLoss
接受类索引的形状,而不是单热编码。
要么适当地构建你的标签张量,要么使用它stat_batch.argmax(axis=1)
。
推荐阅读
- reactjs - 在将数据发布到 API 之前,将日期格式化为“YYYY-MM-DD”,Redux-Saga
- ios - Xamarin.iOS 应用在 Testflight 上的 iOS13 上启动时崩溃
- gremlin - Gremlin 查询以选择实体中的公共属性
- ms-access - MS Access 应用程序需要登录用户/密码
- xml - XML 中的 XPath 包含具有两个值的元素
- php - 如何在wordpress上传递通过CF7提交的表单的字段值?
- javascript - 我应该如何声明一个具有 setInterval 值的变量?
- c - 如何在 vscode 中正确包含带有 msvc 的头文件?
- excel - 如何复制带有空白单元格的列并粘贴到新的工作表 VBA 中?
- extjs - 如何在 ExtJs 中绑定监听器?