diff --git a/Network.py b/Network.py
index 3be278a..7727bc1 100755
--- a/Network.py
+++ b/Network.py
@@ -10,7 +10,7 @@
 from keras.models import Sequential
 from keras.layers.convolutional import Convolution2D, ZeroPadding2D, AveragePooling2D
 from keras import backend as K
-
+K.set_image_dim_ordering('th')
 
 #### There are 3 Parts #####
 
@@ -334,4 +334,4 @@ def grads(self, x):
     imsave(fname, img)
     end_time = time.time()
     print('Image saved as', fname)
-    print('Iteration %d completed in %ds' % (i+1, end_time - start_time))
\ No newline at end of file
+    print('Iteration %d completed in %ds' % (i+1, end_time - start_time))