Can you release model with smaller input size?
seedlingfl opened this issue · 2 comments
seedlingfl commented
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 commented
Is the model size the problem?
seedlingfl commented
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.