decoder.model.0.conv_block.1.weight torch.Size([512, 512, 3, 3])
decoder.model.0.conv_block.2.weight torch.Size([512])
decoder.model.0.conv_block.2.bias torch.Size([512])
decoder.model.0.conv_block.2.running_mean torch.Size([512])
decoder.model.0.conv_block.2.running_var torch.Size([512])
decoder.model.0.conv_block.2.num_batches_tracked torch.Size([])
decoder.model.0.conv_block.5.weight torch.Size([512, 512, 3, 3])
decoder.model.0.conv_block.6.weight torch.Size([512])
decoder.model.0.conv_block.6.bias torch.Size([512])
decoder.model.0.conv_block.6.running_mean torch.Size([512])
decoder.model.0.conv_block.6.running_var torch.Size([512])
decoder.model.0.conv_block.6.num_batches_tracked torch.Size([])