Add head upsampling to UNetPlusPlus decoder

isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.

MIT License

98 stars 8 forks source link

Add head upsampling to UNetPlusPlus decoder #48

Closed tayden closed 3 months ago

tayden commented 3 months ago

The UnetPlusPlus model is seemingly missing the head_upsampling initialization parameter. This PR adds it (copying the documentation and style from the UNet decoder)

isaaccorley commented 3 months ago

Thanks for the PR! Tests are failing for some reason but will merge without them because it's caused by a bug in the actions.