I was working on the week-3 assignment:
This is the ‘unet_model’ block:
cblock1 = conv_block(inputs,n_filters)
cblock2 = conv_block(cblock1[0], n_filters2)
cblock3 = conv_block(cblock2[0], n_filters4)
cblock4 = conv_block(cblock3[0], n_filters8, dropout_prob=0.3)
cblock5 = conv_block(cblock4[0], n_filters8, dropout_prob=0.3, max_pooling=False)
ublock6 = upsampling_block(cblock5[0], cblock4[1], n_filters8)
ublock7 = upsampling_block(ublock6, cblock3[1], n_filters8)
ublock8 = upsampling_block(ublock7, cblock2[1], n_filters4)
ublock9 = upsampling_block(ublock8, cblock1[1], n_filters2)
This should be correct according to my understanding, However it gives an error:
Test failed
Expected value
[‘Conv2D’, (None, 6, 8, 512), 1180160, ‘same’, ‘relu’, ‘HeNormal’]
does not match the input value:
[‘Conv2D’, (None, 6, 8, 256), 590080, ‘same’, ‘relu’, ‘HeNormal’]
Can you tell me where I went wrong?