updated to work with Keras 2.3.1

J13253902839 · Oct 24, 2019 · 728989b · 728989b
1 parent de397ad
commit 728989b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wide_resnet.py b/wide_resnet.py
@@ -23,7 +23,7 @@ def __init__(self, image_size, depth=16, k=8):
         self._use_bias = False
         self._weight_init = "he_normal"
 
-        if K.image_dim_ordering() == "th":
+        if K.image_data_format() == "channels_first":
             logging.debug("image_dim_ordering = 'th'")
             self._channel_axis = 1
             self._input_shape = (3, image_size, image_size)