Hello everyone in the community,
I am training a point cloud classification model using PyTorch. I want to run this custom model on the Hailo8L AI kit, so I am trying to convert it to ONNX format. However, when I attempt to convert the ONNX model to Hailo’s HAR format, I encounter the following error:
Parsing failed with recommendations for end node names: ['/Relu_1']
I am using the following code to convert my PyTorch model to ONNX:
import torch
import onnx
from model_utils import PointCloudModel
model_path = "Training_results_5/Training_results_5_model.pth"
model = PointCloudModel()
inp = [torch.randn((1, 3, 64), requires_grad=False)]
model(*inp)
onnx_path = 'small_example.onnx'
torch.onnx.dynamo_export(model, tuple(inp), onnx_path,
export_params=True,
training=torch.onnx.TrainingMode.PRESERVE,
do_constant_folding=False,
opset_version=13)
I have tried different conversion methods, but I still encounter various errors. Could you help me understand the cause of this issue and how to resolve it?
Best regards