Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Export trained model to onnx and Latency of network #70

Open
John1231983 opened this issue Aug 21, 2023 · 0 comments
Open

Export trained model to onnx and Latency of network #70

John1231983 opened this issue Aug 21, 2023 · 0 comments

Comments

@John1231983
Copy link

Thanks for the great library for hardware-aware NAS. I trained ProxylessNas with the CIFAR dataset. Here are the steps:

Step 1:

python main.py experiment=classification/mobilenet/cifar10_search_latency

Step 2:

python main.py experiment=classification/mobilenet/cifar10_train_latency

I have three questions:

  1. After training the model (step 2), how can I export the trained model to ONNX?
  2. Regarding latency, how can I set a target latency so that the latency of the trained model is smaller than the expected latency?
  3. The latency used on the repo that is on the host device, how can I perform a search with latency constraints based on the latency observed on the target/edge device?
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

No branches or pull requests

1 participant