Is it possible to provide the pytorch model and weights?

[boyang9602]

Hi,
I am doing some research work on the deep learning models in ADS. I am wondering if it is possible to provide the PyTorch model structure and weights instead of just the ONNX models. It is very helpful for doing some modification, training, or fine-tuning.

I am currently working on TLR-related tests. For the SSD-based TLR, I found that the model repo is linked and the weights file is provided, however, I failed to load the model weights with the linked model structure repo (https://github.com/orgs/autowarefoundation/discussions/3907). I also found that the ONNX model is provided and I could load and run it.
For YOLOX-based TLR, I can only find the ONNX models.

It is useful to deploy using ONNX models, however, it is not convenient to modify the model structure, train or fine-tune the model. I managed to convert them to PyTorch models with some tools, however, it is still not easy to modify the model structure.

Thank you for any help!