zoukankan      html  css  js  c++  java
  • 记录报错

    错误1:

    Expected object of backend CPU but got backend CUDA for argument #2 'weight'   

    #之前:
        model = model.cuda()
    #之后:
        model = torch.nn.DataParallel(model).cuda()

    错误2:

    IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

     #之前:
    # 前向传播计算输出
                outputs = model(input)
                label = label.squeeze()
                outputs.detach_()
                # 计算准确率,记录损失
                loss = criterion(outputs[i], label)
    #之后:
                 loss = criterion(outputs, label)            
    #之前label有一个维度问题 
    label.size()=tensor([16,1])
    
    label = label.squeeze()
    
    label.size()=tensor([16])
  • 相关阅读:
    my eye
    html与HTML5的区别
    h5css样式
    h5css3弹性盒子
    简单js的介绍
    2020.8.16(周报6)
    2020.8.18
    2020.8.20
    2020.8.17
    2020.8.15
  • 原文地址:https://www.cnblogs.com/Manuel/p/13267595.html
Copyright © 2011-2022 走看看