0

我試圖培養一個非常基本的CNN上CIFAR10數據集,並得到以下錯誤: AttributeError的:「CrossEntropyLoss」對象有沒有屬性「落後」AttributeError的:「CrossEntropyLoss」對象有沒有屬性「落後」

criterion =nn.CrossEntropyLoss 
optimizer=optim.SGD(net.parameters(),lr=0.001,momentum=0.9) 

for epoch in range(2): # loop over the dataset multiple times 
     running_loss = 0.0 
     for i, data in enumerate(trainloader, 0): 
      # get the inputs 
      inputs, labels = data 
      # wrap them in Variable 
      inputs, labels = Variable(inputs), Variable(labels) 

      # zero the parameter gradients 
      optimizer.zero_grad() 
      # forward + backward + optimize 
      outputs = net(inputs) 
      loss = criterion(outputs, labels) 
      loss.backward() 
      optimizer.step() 
      # print statistics 
      running_loss += loss.data[0] 
      if i % 2000 == 1999: # print every 2000 mini-batches 
       print('[%d, %5d] loss: %.3f' % 
         (epoch + 1, i + 1, running_loss/2000)) 
       running_loss = 0.0 

回答

0

問題已解決。我的錯誤,我遺漏了圓括號

criterion =nn.CrossEntropyLoss() 
相關問題