how to i use psconv in my own model
yxt132 opened this issue · 3 comments
yxt132 commented
This is very interesting work. I am tempted to give it a try on my own model for segmentation. Can you provide some instructions on how to replace a standard conv with psconv in my model?
d-li14 commented
Thanks to the compact characteristic of PSConv, just replace nn.Conv2d
with PSConv2d
. Note that there exists another hyperparameter named parts
you may set in our PSConv operator.
yxt132 commented
Cool! Thanks for the speedy response!
I was able to make it work. What's your recommended setting for "parts".
d-li14 commented
It is 4 as the default argument.