RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘（实测百分百有效）

报错

RuntimeError: "unfolded2d_copy" not implemented for 'Half'

原因

模型传入的参数use_half=true，就是利用fp16混合精度计算对CPU进行推理，使用fp16来加快速度但是pytorch cpu不支持fp16，

解决方法

将use_half=False
或者将half()修改为float()

这样模型就能进行计算了；

我报错的修改：

希望这篇文章对你有用！

谢谢点赞评论！

RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘（实测百分百有效）相关推荐

YOLOv5+deepsort——报错解决：RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘
一.报错原因 CPU不支持半精度训练二.解决办法要么采用GPU训练要么将检测模块中的.half()全部改成.float()(计算时请保证数据类型一致)
[debug] RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘
问题描述在使用nn.CrossEntropyLoss时出现报错: loss_id = self.IDLoss(id_preds, id_targets) Traceback (most recent ...
RuntimeError: “LayerNormKernelImpl“ not implemented for ‘Long‘
将GitHUB上下载的模块用于自己的模型,一定要注意输入数据的数据类型(dtype)是否满足模块的要求第一步:数据类型为张量:train_inputs=torch.tensor(train_inpu ...
教你提高时间管理和知识管理水平，实现职场高效率
要想职场效率高,掌握一些技巧肯定是不能少的,对于同一份工作,别人用10分钟完成了,而你用了一个小时依然没有搞完,那么别人可以比你享受更多的空闲时刻,而你在繁忙中把一天的时间给过完了,但是你发现自己看似 ...
pytorch-yolov5_deepsort目标跟踪行人车辆计数
看图就可以知道.穿过两条线,分别用于测上行人流和下行人流. 这个cpu和gpu版本都可以跑,cpu的话,安装好相应的库之后,运行会报 RuntimeError: "unfolded2d_co ...
PyTorch 笔记（09）— Tensor 线性代数计算（torch.trace、torch.diag、torch.mm、torch.dot、torch.inverse逆矩阵、转置）
1. 常用函数常用线性表函数如下表所示: 2. 使用示例 2.1 torch.trace In [22]: import torch as tIn [23]: a = t.arange(1, 10) ...
timm 视觉库中的 create_model 函数详解
timm 视觉库中的 create_model 函数详解最近一年 Vision Transformer 及其相关改进的工作层出不穷,在他们开源的代码中,大部分都用到了这样一个库:timm.各位炼丹师 ...
Web接收实时视频流并显示（flask创建路由接收）| 解决延时不同步问题
在开发B/S架构的检测算法时,需要对摄像头实时视频流进行检测处理. 下文代码能实现接收实时视频流到web端指定路由下进行播放.在后续开发中可继续将该路由视频流引入到相应的前端Html文件中的img指定 ...
pytorch实现 vision_transformer
github地址:https://github.com/rwightman/pytorch-image-models/ 记录一下模型,方便查阅 """ Vision Tr ...

RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘（实测百分百有效）

报错

原因

解决方法

我报错的修改：

RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘（实测百分百有效）相关推荐

最新文章

热门文章