Merge pull request allanzelener#7 from AdoHaha/master

changes towards keras 2
HabibMrad · Mar 20, 2017 · 09aff44 · 09aff44
2 parents a85cd36 + a3080a6
commit 09aff44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yad2k.py b/yad2k.py
@@ -19,7 +19,7 @@
 from keras.layers.normalization import BatchNormalization
 from keras.models import Model
 from keras.regularizers import l2
-from keras.utils.visualize_util import plot
+from keras.utils.vis_utils import plot_model as plot
 
 from yad2k.models.keras_yolo import (space_to_depth_x2,
  space_to_depth_x2_output_shape)