Update.

[flatland.git] / flatland.h

diff --git a/flatland.h b/flatland.h
index 734add4..e86533e 100644 (file)
--- a/flatland.h
+++ b/flatland.h
@@ -1,4 +1,5 @@
 
 THByteTensor *generate_sequence(long nb_sequences,
-                                long nb_images_per_sequence,
-                                long image_width, long image_height);
+                                long nb_images,
+                                long image_height, long image_width,
+                                int random_shape_size, int random_colors);