PyTorch Squeeze-SegNet Implementation