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)
可以成功转换