video_grid -> video_grid2d

AgnezIO · Oct 15, 2015 · cf99ca6 · cf99ca6
1 parent b823e29
commit cf99ca6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agnez/keras_callbacks.py b/agnez/keras_callbacks.py
@@ -1,5 +1,5 @@
 '''This provides visualization tools for Keras.'''
-from . import grid2d, DeepPref, video_grid
+from . import grid2d, DeepPref, video_grid2d
 
 from theano import function
 from keras.callbacks import Callback
@@ -153,7 +153,7 @@ def __init__(self, filepath, X, func, how_often=10, display=None):
     def on_epoch_end(self, epoch, logs={}):
         if epoch % self.how_often == 0:
             rec = self.func(self.X)
-            _ = video_grid(rec.transpose(1, 0, 2), self.filepath)
+            _ = video_grid2d(rec.transpose(1, 0, 2), self.filepath)
             if self.display is not None:
                 self.display.clear_output(wait=True)
         print('Saved gif.')