Building a 3D Unet

It seems model.decoder[1].conv1.conv_layer gets a wrong input tensor, which is expected to have 32 channels, but has 48.
I would recommend to print the shapes of all tensors in your forward method, which would make debugging easier.
I guess you might have concatenated two activations with 32 and 16 channels accidentally.