python - 属性错误 : 'CrossEntropyLoss' object has no attribute 'backward'

标签 python neural-network deep-learning conv-neural-network pytorch

我正在尝试在 CIFAR10 数据集上训练一个非常基本的 CNN,但出现以下错误: AttributeError: 'CrossEntropyLoss' 对象没有属性 'backward'

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

最佳答案

问题已解决。我的错误,我漏掉了括号

criterion = nn.CrossEntropyLoss()

关于python - 属性错误 : 'CrossEntropyLoss' object has no attribute 'backward' ,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47488598/

相关文章:

python - 如何将 SQL 标量子查询转换为 SQLAlchemy 表达式

r - Keras R 图像分类模型中的形状错误

c# - C# 中的激活函数列表

computer-vision - Caffe:具有不同数量标签的多标签图像

python - 将 PyCharm 与 Git 一起使用时,我应该忽略 .idea 文件夹吗?

python - 替换 beautifulsoup 中的 <p> 和 <br> 标签

python - 沿多个维度的 Tensorflow argmax

Tensorflow 语义分割给出零损失

python - 遇到 : json.decoder.JSONDecodeError: Expecting value: line 1 column 1 (char 0)

python - 输入到神经网络的 NP 数组会导致形状错误