bpu argmax问题

[Sunseeker]

1、语义分割中，在最后一步加了torch.argmax之后，转出.bin文件后推理结果全为0，请问有类似的情况吗

[颜值即正义]

您好，请问quantized.onnx模型的推理正常吗？目前没遇到过您说的这种情况哎

[Sunseeker]

x = x.argmax(dim=1)-
x = torch.tensor(x, dtype=torch.float32)

在 argmax后转一下dtype就好了

[颜值即正义]

感谢您贴出该问题的解决方案，祝好~一起加油！