@Oli thanks a lot. I did get print of updated model as follows:

down1.conv_block.conv1.weight          torch.Size([48, 1, 3, 3])
down1.conv_block.conv2.weight          torch.Size([48, 48, 3, 3])

down2.conv_block.conv1.weight          torch.Size([96, 48, 3, 3])
down2.conv_block.conv2.weight          torch.Size([96, 96, 3, 3])

down3.conv_block.conv1.weight          torch.Size([192, 96, 3, 3])
down3.conv_block.conv2.weight          torch.Size([192, 192, 3, 3])

down4.conv_block.conv1.weight          torch.Size([384, 192, 3, 3])
down4.conv_block.conv2.weight          torch.Size([384, 384, 3, 3])

down5.conv1.weight          torch.Size([768, 384, 3, 3])
down5.conv2.weight          torch.Size([768, 768, 3, 3])

up4.conv_trans1.weight          torch.Size([1152, 384, 2, 2])
up4.conv3.weight          torch.Size([384, 384, 3, 3])
up4.conv4.weight          torch.Size([384, 384, 3, 3])

up3.conv_trans1.weight          torch.Size([576, 192, 2, 2])
up3.conv3.weight          torch.Size([192, 192, 3, 3])
up3.conv4.weight          torch.Size([192, 192, 3, 3])

up2.conv_trans1.weight          torch.Size([288, 96, 2, 2])
up2.conv3.weight          torch.Size([96, 96, 3, 3])
up2.conv4.weight          torch.Size([96, 96, 3, 3])

up1.conv_trans1.weight          torch.Size([144, 48, 2, 2])
up1.conv3.weight          torch.Size([48, 48, 3, 3])
up1.conv4.weight          torch.Size([48, 48, 3, 3])

out.weight          torch.Size([2, 48, 3, 3])


apparently the number of channel from up4 to up1 isn’t same as pic.