I have been trying to debug the train_step function - I am getting the following error message though I seem to be following all of the steps correctly:
ValueError: in user code:
File "<ipython-input-29-9075c3053067>", line 17, in train_step *
print(tf.shape(a_G))
ValueError: Dimension 1 in both shapes must be equal, but are 50 and 25. Shapes are [1,50,50,512] and [1,25,25,512].
From merging shape 3 with other shapes. for '{{node Shape/packed}} = Pack[N=6, T=DT_FLOAT, axis=0](model/block1_conv1/Relu, model/block2_conv1/Relu, model/block3_conv1/Relu, model/block4_conv1/Relu, model/block5_conv1/Relu, model/block5_conv4/Relu)' with input shapes: [1,400,400,64], [1,200,200,128], [1,100,100,256], [1,50,50,512], [1,25,25,512], [1,25,25,512]. Your help would be greatly appreciated! Thank you.