解决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 111阅读 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 条评论，111人围观）

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

相关阅读

相关 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 赞/ 134 阅读

相关解决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 赞/ 112 阅读

相关 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 赞/ 104 阅读

相关【PyTorch】torch.mean(), dim=0, dim=1 详解

创建一个tensor,这个tensor是一个元素类型为浮点型的2维数组 import torch s = torch.arange(6,dtype=floa

红太狼/ 2022年11月13日 04:10/ 0 赞/ 303 阅读

相关 C - ItoA: Convert a number to a character string (the inverse of AtoI)

分享一个大牛的人工智能教程。零基础！通俗易懂！风趣幽默！希望你也加入到人工智能的队伍中来！请点击[http://www.captainbed.net][http_www.cap

电玩女神/ 2022年11月01日 13:43/ 0 赞/ 17 阅读

相关 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 赞/ 112 阅读

相关 TensorFlow报错：Tried to convert ‘input‘ to a tensor and failed. Error: None values not supported.

尝试给placeholder赋值： input_placeholder = tf.placeholder(tf.int32, [batch_size

曾经终败给现在/ 2022年09月09日 06:58/ 0 赞/ 134 阅读

相关 Python convert a list to String

正常情况如果list由字串类型，一般可以用下面这种方法 a_list = ["1", "2", "3"] print("".join(a_list)) 发散

梦里梦外;/ 2022年06月03日 04:41/ 0 赞/ 152 阅读

相关 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 赞/ 120 阅读

相关 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 赞/ 130 阅读