yolo加模块报错RuntimeError: Input type and weight type should be the same

原创于 2025-03-06 15:05:34 发布 · 575 阅读

11 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #人工智能 #YOLO

部署运行你感兴趣的模型镜像

当yolo增加一些模块的时候训练完第一轮进行验证的时候就报错

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same

主要是混合精度的问题

解决方案：

1、把amp=False

2、到ultralytics/engine/validator.py文件里面 crtl+f搜索half

将

self.args.half = self.device.type != 'cpu'  # force FP16 val during training

注释掉,然后增加一个

self.args.half =False

然后重新训练就ok了，如果跑其他的再改回去

您可能感兴趣的与本文相关的镜像

Yolo-v5

Yolo

YOLO（You Only Look Once）是一种流行的物体检测和图像分割模型，由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出，因其高速和高精度而广受欢迎

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

yw408

关注关注

7
点赞
踩
11

收藏

觉得还不错? 一键收藏
0
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
打赏
打赏
打赏举报

举报

报错处理：RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should...

cnjs1994的博客

04-11

4478

这篇博客解决的是pytorch训练图像分类模型中常常遇到的一个常见问题：就是模型在GPU,但是数据加载到了CPU

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the

程序员，他们想的是什么?他们想的永远都是技术，他们崇尚的也永远都是技术。

11-02

623

使用PyTorch可视化工具-Tensorboard；writer.add_graph(model,graph_inputs)#输出模型结构报错如下。

参与评论您还未登录，请先登录后发表或查看评论

【PyTorch】常见错误: RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)

热门推荐

Raywit的博客

06-30

7万+

【PyTorch】常见错误错误： RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 问题原因：错误内容就在类型不匹配，根据报错内容可以看出Input type为

RuntimeError_ Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)

一起加油~

07-06

2万+

RuntimeError:输入类型()和权重类型()应该相同，或者输入应该是一个MKLDNN张量，而权重是一个密集张量。

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)

weixin_44012667的博客

07-26

566

【PyTorch】常见错误错误：RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 问题原因：错误内容就在类型不匹配，根据报错内容可以看出Input type为_input type (torch.floattensor) and weight type (torch.cuda.floattensor) shou。

YOLO RuntimeError: Given groups=1, weight of size [128, 256, 1, 1], expected input[1, 512, 16, 16] to have 256 channels, but got 512 channels instead

03-25

Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! ``` 要解决此问题，可以确保所有的张量都位于同一个设备上。可以通过 `.to(device)` 方法将所有涉及的张量移动...

YOLOv8改进时遇到的报错及其解决

xzgxg1103的博客

02-19

548

解决：通过--master_port参数来指定主节点使用的端口号如：CUDA_VISIBLE_DEVICES=4,5 torchrun --nproc_per_node=2 --master_port 15666 train_test.py。torchvision.transforms.functional_tensor 替换成torchvision.transforms.functional 就能正常运行了。在服务器环境中，特别是当多个用户或任务同时运行时，可能已经被其他DDP训练任务占用。

YOLOv10模型应用过程中的报错处理及疑问解答，涉及环境搭建、模型训练、模块改进、论文写作

Limiiiing的博客

11-07

703

本文记录了YOLOv10模型的这一系列过程中可能发生的问题，以及大家非常关心的问题，这里统一做了一个记录和解答。

Error: DWT did not return enough components. Traceback (most recent call last): File "E:\bug\test.py", line 3, in <module> model = YOLO(r"E:\bug\ultralytics\cfg\models\11\yolo11-WTConv.yaml") File "E:\bug\ultralytics\models\yolo\model.py", line 23, in __init__ super().__init__(model=model, task=task, verbose=verbose) File "E:\bug\ultralytics\engine\model.py", line 143, in __init__ self._new(model, task=task, verbose=verbose) File "E:\bug\ultralytics\engine\model.py", line 251, in _new self.model = (model or self._smart_load("model"))(cfg_dict, verbose=verbose and RANK == -1) # build model File "E:\bug\ultralytics\nn\tasks.py", line 337, in __init__ m.stride = torch.tensor([s / x.shape[-2] for x in _forward(torch.zeros(1, ch, s, s))]) # forward File "E:\bug\ultralytics\nn\tasks.py", line 335, in _forward return self.forward(x)[0] if isinstance(m, (Segment, Pose, OBB)) else self.forward(x) File "E:\bug\ultralytics\nn\tasks.py", line 113, in forward return self.predict(x, *args, **kwargs) File "E:\bug\ultralytics\nn\tasks.py", line 131, in predict return self._predict_once(x, profile, visualize, embed) File "E:\bug\ultralytics\nn\tasks.py", line 152, in _predict_once x = m(x) # run File "D:\Program\anaconda\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 1751, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "D:\Program\anaconda\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 1762, in _call_impl return forward_call(*args, **kwargs) File "E:\bug\ultralytics\nn\modules\conv.py", line 96, in forward y_LL = self.conv(Y_LL) File "D:\Program\anaconda\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 1751, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "D:\Program\anaconda\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 1762, in _call_impl return forward_call(*args, **kwargs) File "D:\Program\anaconda\envs\torch\lib\site-packages\torch\nn\modules\conv.py", line 554, in forward return self._conv_forward(input, self.weight, self.bias) File "D:\Program\anaconda\envs\torch\lib\site-packages\torch\nn\modules\conv.py", line 549, in _conv_forward return F.conv2d( RuntimeError: Input type (__int64) and bias type (float) should be the same

最新发布

07-30

在使用YOLO模型时，若遇到 `Error DWT did not return enough components` 和 `RuntimeError: Input type (__int64) and bias type (float) should be the same` 错误，通常涉及两个问题：一是小波变换（DWT）返回的...

YOLOv8模型应用过程中的报错处理及疑问解答，涉及环境搭建、模型训练、模块改进、论文写作

Limiiiing的博客

12-25

640

本文记录了YOLOv8模型的这一系列过程中可能发生的问题，以及大家非常关心的问题，这里统一做了一个记录和解答。

Pytorch出现RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)

maohule的博客

07-18

1985

上面四行任选一，还有其他未列出的表述方法，都可以将模型加载到GPU。有时已经写了将模型和输入数据加载到GPU的语句，但还是报了这个错误。问题原因搞清楚了，模型没加载到CPU，在代码中加一行语句就可以了。这个函数功能是产生一个新的tensor，并不会改变原数据。问题来源是输入数据没有加载到GPU，解决方法为。是一个“in-place”方法，从报错问题描述中可以找到错误原因。

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the

P13643822101的博客

08-13

1万+

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the

【BUG解决】 RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)

Gorgeous_wxd

12-13

4780

问题描述 RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same 解决方法对net或者新添加的模块进行转换。解决方法1 device = torch.device('cuda:0') net= net.to(device) 解决方法2 net = net.cuda() ...

【BUG解决】RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)

Gorgeous_wxd

12-13

2547

问题描述 RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 解决方法方法1 device = torch.device('cuda:0') inputs = inputs.to(device) 方法2 inputs = inputs.cuda()

pytorch参数转不到cuda，RuntimeError: Input type (torch.cuda.FloatTensor) and weight type（torch.FloatTensor

裂缘冰释

10-31

592

pytorch模型参数（卷积）转不到cuda 错误提示：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type（torch.FloatTensor）

runtimeerror: input type (torch.cuda.floattensor) and weight type (torch.floattensor) should be the

weixin_48867081的博客

07-07

954

在 # Load model device = select_device(device) model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data, fp16=half) model.cuda()#使用gpu，用这个将模型数据导入，如用cpu跑把这段注释即可 stride, names, pt = model.stride, model.names, model.pt 说明：加入model.cuda() 或者model =

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be

weixin_42314144的博客

03-26

1209

训练yolov8时遇到该问题，debug时conv.py出现return F.conv2d(input, weight, bias, self.stride,报错。在forward里添加一行，输入(input)张量的数据类型转换为与权重(weight)张量相同的数据类型.