d-li14/PSConv

how to i use psconv in my own model

yxt132 opened this issue · 3 comments

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?

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.

Cool! Thanks for the speedy response!

I was able to make it work. What's your recommended setting for "parts".

It is 4 as the default argument.