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

Add TorchScript version #76

Merged
merged 2 commits into from
Mar 5, 2021
Merged

Add TorchScript version #76

merged 2 commits into from
Mar 5, 2021

Conversation

yarkable
Copy link
Contributor

@yarkable yarkable commented Feb 27, 2021

Hi, I just did something that can export PyTorch model to TorchScript version

  1. Add modified MODNet that satisfies TorchScript version
  2. Modify MobileNetV2 (Due to torch.jit doesn't support lambda expression)
  3. Upload an exported TorchScript model using official pretrain model (See README.md for detail)

@ZHKKKe
Copy link
Owner

ZHKKKe commented Feb 27, 2021

Hi, thanks for your contributions! I will test the code and merge it next week. Then I will give you a feedback.

@yarkable
Copy link
Contributor Author

Thanks a lot !

@ZHKKKe ZHKKKe merged commit 26cae0f into ZHKKKe:master Mar 5, 2021
ZHKKKe added a commit that referenced this pull request Mar 5, 2021
@ZHKKKe
Copy link
Owner

ZHKKKe commented Mar 5, 2021

@yarkable
Hi, I have tried to merge you branch. However, it seems that the code will affect some other demos. Therefore, I reversed it.
Could you start a new pull requests to the branch develop?
In this way, I can process it before merge it to master. Thanks.

@yarkable
Copy link
Contributor Author

yarkable commented Mar 5, 2021

No problem.

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