When running the train step function, I am running into a dimension mistmatch issue. This happens when I call the compute_style_cost function. Needless to say that the unit tests up to this exercise have passed. The error is thrown when the gram matrix function is being called in the compute_layer_style_cost function.
I have a feeling that this has to do with how I reshaped a_S and a_G in the compute_layer_style_cost and specifically how I handle (or rather dont handle) the ‘m’ dimension. I further think it has to do with how I use tf.reshape.
I have tried to implement what is in the instructions and read numerous other posts but cannot really identify what is different in my implementation.
The error I am at is: ValueError: Dimensions must be equal, but are 64 and 400 for ‘{{node MatMul}} = BatchMatMulV2[T=DT_FLOAT, adj_x=false, adj_y=false](transpose_1/x, transpose_2)’ with input shapes: [1,400,400,64], [64,400,400,1].