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

Can you release model with smaller input size? #58

Open
seedlingfl opened this issue Oct 24, 2023 · 2 comments
Open

Can you release model with smaller input size? #58

seedlingfl opened this issue Oct 24, 2023 · 2 comments
Labels
enhancement New feature or request question Further information is requested

Comments

@seedlingfl
Copy link

The input size 1024x1024 uses too much memory for my case, would you consider release models with smaller input size, such as 512x512?
Or directly convert your pretrained model (pytorch?) to onnx using 512x512 input size might also work well.

With the static onnx model, it seems very hard to change the input size.

@DanielHauschildt DanielHauschildt added enhancement New feature or request question Further information is requested labels Nov 6, 2023
@DanielHauschildt
Copy link
Contributor

Is the model size the problem?

@seedlingfl
Copy link
Author

Actually I am able to use the 1024x1024 model in my case now. Thanks.
But it will still be good if there is option to use smaller input size to reduce memory usage.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants