Course Four Week 4 Final Lab (Art Generation with Neural Style Transfer)

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.

To help me assist you,

Please click on my name to start a private message. Then, attach your notebook as a .ipynb file. Please note that mentors cannot access your Coursera Jupyter workspace, so sending the notebook in a .ipynb format is essential.