U-net model from https://arxiv.org/abs/1505.04597 implemented with additional features:
Requires PyTorch 0.4