pytorch-unet This model is a U-Net with a configurable depth. Sometimes having the default depth just isn't enough! You will need PyTorch version >= 0.3.0 and TorchVision version >= 0.2.0