The UNet definition you’ve linked uses a defined number of input and output channels for the convolution blocks. You could simply divide all in and out channels by 2 to reduce the number of parameters.