Skip to content

Conversation

Phoenix8215
Copy link

@Phoenix8215 Phoenix8215 commented Sep 25, 2024

Summary

This PR adds support for the GhostNet V1, V2, and V3 models in the PyTorchx repository. These models implement the GhostNet architecture, which is efficient for mobile and edge devices.

Changes

  • Added ghostnetv1, ghostnetv2, and ghostnetv3 models.
  • Verified model integrity using torch.onnx.export and pytorch-summary tools.

Test and Validation

  • Exported ONNX format and simplified using onnx-simplifier.
  • Ran inference to confirm the correct output size and behavior.
  • Tested with PyTorch 1.3.0, CUDA 10.0, Python 3.7.3 as mentioned in the environment setup.

Future Work

  • 🚀Plan to extend support for GhostNet series to the tensorrtx repository to enable TensorRT acceleration.

@wang-xinyu
Copy link
Owner

@Phoenix8215

Can you please add a README.md in the ghostnet folder?

Thanks for your contribution!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants