导出onnx模型报错

[dengfenglai321]

大佬您好，我在导出onnx模型时有fail，没有成功导出onnx模型文件，如下：
UserWarning: ONNX export failed on threshold because non-zero threshold not supported
请问有办法解决吗

[SWHL]

@cendelian 我也遇到同样问题，请问你解决了吗

[TD-wzw]

@SWHL RuntimeError: ONNX export failed: Couldn't export operator aten::threshold
我是这样的错误请问大佬您解决了吗？

[SWHL]

@SWHL RuntimeError: ONNX export failed: Couldn't export operator aten::threshold
我是这样的错误请问大佬您解决了吗？

• 解决了
• 将以下函数
  

  DBNet.pytorch/models/backbone/MobilenetV3.py

  Lines 20 to 24 in 678b2ae

  	def forward(self, x):
  	x = (self.slope * x) + self.offset
  	x = F.threshold(-x, -1, -1)
  	x = F.threshold(-x, 0, 0)
  	return x

• 更改为:

  def forward(self, x):
      x = (self.slope * x) + self.offset

      x = torch.clamp(-x, min=-1)
      x = torch.clamp(-x, min=0)
      return x

[TD-wzw]

@SWHL 运行时错误：ONNX 导出失败：无法导出操作符 aten::threshold
我是这样的错误请问各位大佬解决了吗？

• 解决了
• 将以下函数
  

  DBNet.pytorch/models/backbone/MobilenetV3.py

  Lines 20 to 24 in 678b2ae

  	def forward(self, x):
  	x = (self.slope * x) + self.offset
  	x = F.threshold(-x, -1, -1)
  	x = F.threshold(-x, 0, 0)
  	return x

• 更改为：

  DEF 向前（自，X）：
       X  =（自我。斜率 *  X）+ 自我。偏移量

      x  = 火炬。夹具( - x , min = - 1 )
       x  = 割炬。钳位( - x , min = 0 )
      返回 x

感谢，转成功了，大佬你试过onnx-simplifier然后转ncnn吗？

[SWHL]

@SWHL 运行时错误：ONNX 导出失败：无法导出操作符 aten::threshold
我是这样的错误请问各位大佬解决了吗？

• 解决了
• 将以下函数
  

  DBNet.pytorch/models/backbone/MobilenetV3.py

  Lines 20 to 24 in 678b2ae

  	def forward(self, x):
  	x = (self.slope * x) + self.offset
  	x = F.threshold(-x, -1, -1)
  	x = F.threshold(-x, 0, 0)
  	return x

• 更改为：

  DEF 向前（自，X）：
       X  =（自我。斜率 *  X）+ 自我。偏移量

      x  = 火炬。夹具( - x , min = - 1 )
       x  = 割炬。钳位( - x , min = 0 )
      返回 x

感谢，转成功了，大佬你试过onnx-simplifier然后转ncnn吗？

这个没有试过了