[解决] We don‘t have an op for aten::normal but it isn‘t a special case

news/2024/11/18 3:33:18/

pytorch 网络转为 onnx时遇到的问题:

错误

The error is :
RuntimeError: 0 INTERNAL ASSERT FAILED at “/pytorch/torch/csrc/jit/ir/alias_analysis.cpp”:532, please report a bug to PyTorch. We don’t have an op for aten::normal but it isn’t a special case. Argument types: Tensor, float, float, None

意思:
运行时错误:0 内部断言失败在“/pytorch/torch/csrc/jit/ir/alias_analysis.cpp”:532,请向 PyTorch 报告错误。没有 aten::normal 的操作,但这不是一个特例。参数类型:张量、浮点数、浮点数、None

系统环境

OS: Ubuntu 18.04.6 LTS
Python version: 3.7 (64-bit runtime)
Is CUDA available: True
CUDA runtime version: 11.1.74
GPU models and configuration: GPU 0: NVIDIA GeForce RTX 3090 Ti
Nvidia driver version: 510.108.03

出错原因

原始代码中需要生成一个和输入相同的均值为0,方差为1的标准正态分布,使用了normal_()

eps = torch.FloatTensor(std.size()).normal_()

替换为下面这句,能达到同样的功能

eps = torch.randn_like(std)

可以成功转换


http://www.ppmy.cn/news/4906.html

相关文章

SpringBoot实战(十)集成多数据源dynamic-datasource

目录1.Maven依赖2. DS注解3.普通Hihari连接池3.1 yml配置4.Druid连接池4.1 Druid依赖4.2 yml配置4.3 排除原生的Druid配置5. UserController6. UserServiceImpl7.测试7.1 新增数据7.2 查询数据7.3 测试结果8.源码地址:dynamic-datasource-spring-boot-starter 是一个…

北京理工大学汇编语言复习重点

汇编是半开卷,可以带纸质资料。理论上,学好了以后,带本书进去就ok了,但是这次是线上,我还没书,就对着考试重点整理一点资料用于打印吧。 因为是线上,所以第4章基本不考框架了,浮点操…

PolarDB-X源码解读:DDL的一生(下)

概述 在《DDL的一生(上)》中,我们以添加全局二级索引为例,从DDL开发者的视角介绍了如何在DDL引擎框架下实现一个逻辑DDL。在本篇,作者将从DDL引擎的视角出发,向读者介绍DDL引擎的架构、实现,以…

Sulfo CY5-MAL|磺基-CY5 马来酰亚胺

Sulfo CY5-MAL|磺基-CY5 马来酰亚胺 英文名称:Cyanine5 maleimide Cyanine5 MAL Cy5 maleimide Cy5 MAL CAS:1437872-46-2 外观:深蓝色粉末 分子量:641.24 分子式:C38H45ClN4O3 花菁染料,一种发…

Linux下/dev和/sys/class初探

文章目录引言1. devfs2. sysfssysfs 和 /dev的关系3. procATTR 与 file_operations参考链接:扩展知识引言 在用户空间操控设备可以通过/dev下的设备文件,也可以通过/sys/class下的设备文件。两者到底有什么区别?学习过程中,对于内…

企业级DevOps容器云平台流水线综合解决方案详解(二)

十、Kubeadm快速部署K8s集群 kubeadm是官方社区推出的一个用于快速部署 kubernetes 集群的工具,这个工具能通过两条指令完成一个kubernetes集群的部署。 环境要求: 一台或多台机器,操作系统CentOS 7.x-86_x64; 硬件配置:内存2GB或2G+,CPU 2核或CPU 2核+; 集群内各个机…

【电气安全】关于医疗专用隔离电源系统在医院配电系统中的应用

【摘要】 根据医疗场所分类说明医院尤其是 2 类 医疗场所的接地方案,分析医疗场所自动切断电源措 施,提出校验医疗 IT 系统配出回路长度;说明 RCD、RCM、IMD、IFLS 等的用途和范围;就应急医 院 2 类场所的配电方案进行探讨。 【…

go语言并发编程(中)

go语言并发编程中并发安全和锁互斥锁Mutex读写锁信号量&条件变量单例模式&time包方法协程池并发安全和锁 互斥锁Mutex 有时候在Go代码中可能会存在多个goroutine同时操作一个资源(临界区),这种情况会发生竞态问题(数据竞…