diff --git a/255_256_SRGAN/SRGAN_train.py b/255_256_SRGAN/SRGAN_train.py
index ee89f66e7..833382f68 100644
--- a/255_256_SRGAN/SRGAN_train.py
+++ b/255_256_SRGAN/SRGAN_train.py
@@ -27,6 +27,9 @@
 from keras.layers import UpSampling2D, LeakyReLU, Dense, Input, add
 from tqdm import tqdm
 
+# some versions of Tf.keras will need you to configure channels last to make this notebook work
+# keras.backend.set_image_data_format('channels_last')
+
 #########################################################################
 
 #Define blocks to build the generator
@@ -345,4 +348,4 @@ def create_comb(gen_model, disc_model, vgg, lr_ip, hr_ip):
 plt.title('Orig. HR image')
 plt.imshow(sreeni_hr[0,:,:,:])
 
-plt.show()
\ No newline at end of file
+plt.show()