博客
关于我
pytorch loss = loss_func(output, label) 报错
阅读量:313 次
发布时间:2019-03-03

本文共 424 字,大约阅读时间需要 1 分钟。

在运行损失函数时 loss = loss_func(output, label) 时报了两个错 : 

 

1.

RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'target'

这是因为label必须是 LongTensor 类型, 之前是 :

label = t.tensor(float(image[1]))  # image[1]为str类型

改为 :

label = t.tensor(float(image[1])).long()

 

2. 

Assertion `cur_target >= 0 && cur_target < n_classes’ failed

这是因为我的数据集类别是从1开始,需要依次减一,所以要改为 :

label = t.tensor(float(image[1])-1).long()

 

 

转载地址:http://wncq.baihongyu.com/

你可能感兴趣的文章
Netty客户端断线重连实现及问题思考
查看>>
Netty工作笔记0001---Netty介绍
查看>>
Netty工作笔记0002---Netty的应用场景
查看>>
Netty工作笔记0003---IO模型-BIO-Java原生IO
查看>>
Netty工作笔记0004---BIO简介,介绍说明
查看>>
Netty工作笔记0005---NIO介绍说明
查看>>
Netty工作笔记0006---NIO的Buffer说明
查看>>
Netty工作笔记0007---NIO的三大核心组件关系
查看>>
Netty工作笔记0008---NIO的Buffer的机制及子类
查看>>
Netty工作笔记0009---Channel基本介绍
查看>>
Netty工作笔记0010---Channel应用案例1
查看>>
Netty工作笔记0011---Channel应用案例2
查看>>
Netty工作笔记0012---Channel应用案例3
查看>>
Netty工作笔记0013---Channel应用案例4Copy图片
查看>>
Netty工作笔记0014---Buffer类型化和只读
查看>>
Netty工作笔记0015---MappedByteBuffer使用
查看>>
Netty工作笔记0016---Buffer的分散和聚合
查看>>
Netty工作笔记0017---Channel和Buffer梳理
查看>>
Netty工作笔记0018---Selector介绍和原理
查看>>
Netty工作笔记0019---Selector API介绍
查看>>