Add head upsampling to UNetPlusPlus decoder

isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.

MIT License

104 stars 8 forks source link

Add head upsampling to UNetPlusPlus decoder #48

Closed tayden closed 4 months ago

tayden commented 4 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 4 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.