Hi,



 Janine:

*self._create_layer_2(128, 256),


This is a 2-D convolutional block. Convolutional Blocks perform convolutions and need inputs specifically in the shape of [batch_size,channels,img_height,img_width]. You can see that these are 4-D tensors. You are trying to pass 2-D tensors into the conv block–and getting an error.