解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Pyth

灰太狼 2022-12-31 12:21 147阅读 0赞

### 解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number ###

#### 报错代码 ####

running_loss += loss.data[0] * inputs.size(0)

#### 报错信息 ####

IndexError                                Traceback (most recent call last)
    <ipython-input-8-7f128a52c8c7> in <module>
         19 # Decay LR by a factor of 0.1 every 7 epochs
         20 exp_lr_scheduler = lr_scheduler.StepLR(optimizer_ft, step_size=7, gamma=0.1)
    ---> 21 model_ft = train_model(model_ft, criterion, optimizer_ft, exp_lr_scheduler, num_epochs=1)
         22 
         23 visualize_model(model_ft)
    
    <ipython-input-6-e165c8ed70ce> in train_model(model, criterion, optimizer, scheduler, num_epochs)
         57 
         58                 # statistics
    ---> 59                 running_loss += loss.data[0] * inputs.size(0)
         60                 running_corrects += torch.sum(preds == labels.data)
         61 
    
    IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

#### 解决办法 ####

running_loss += loss.item() * inputs.size(0)

> > > #### 欢迎大家交流学习，任何问题都可以留言 ####

发表评论取消回复

表情：

评论列表（有 0 条评论，147人围观）

还没有评论，来说两句吧...

相关阅读

相关解决(wait)KeyError: “The name ‘image_tensor:0‘ refers to a Tensor which does not exist. The operatio

目录解决(Wait)KeyError: "The name 'image\_tensor:0' refers to a Tensor which does not exis

清疚/ 2024年02月21日 08:46/ 0 赞/ 54 阅读

相关 IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python

pytorch报错： IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a

心已赠人/ 2023年06月25日 10:16/ 0 赞/ 36 阅读

相关 Cannot convert a symbolic Tensor to a numpy array

使用TF进行计算，出现： Traceback (most recent call last): File "infer.py", line 14, in

柔情只为你懂/ 2023年01月17日 09:23/ 0 赞/ 172 阅读

相关解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Pyth

解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim ten

灰太狼/ 2022年12月31日 12:21/ 0 赞/ 148 阅读

相关 IndexError: too many indices for tensor of dimension 0

源代码： import torch import torchvision import torch.nn as nn from model i

小咪咪/ 2022年12月18日 04:57/ 0 赞/ 138 阅读

相关【问题记录】pytorch自定义数据集 No such file or directory， invalid index of a 0-dim

保存模型：：保存整个神经网络的结构和模型参数 torch.save(mymodel, ‘mymodel.pkl’) 只保存神经网络的模型参数 torch

待我称王封你为后i/ 2022年11月04日 14:28/ 0 赞/ 40 阅读

相关 can‘t convert cuda:0 device type tensor to numpy

can't convert cuda:0 device type tensor to numpy 测试代码： import torch i

╰半夏微凉°/ 2022年10月13日 02:00/ 0 赞/ 8 阅读

相关 can‘t convert cuda:0 device type tensor to numpy

can't convert cuda:0 device type tensor to numpy x是list，list内容是tensor cuda 代码： x = np

小鱼儿/ 2022年09月10日 06:14/ 0 赞/ 147 阅读

相关 How to convert a Drawable to a Bitmap?

原文：[http://stackoverflow.com/questions/3035692/how-to-convert-a-drawable-to-a-bitmap][ht

叁歲伎倆/ 2022年05月21日 04:12/ 0 赞/ 170 阅读

相关 TypeError: can‘t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory fi

运行程序如下： import numpy as np import torch from torch import nn from torch

迈不过友情╰/ 2022年05月08日 02:40/ 0 赞/ 186 阅读