Seems to work quite well.

[flatland.git] / mylib.c

diff --git a/mylib.c b/mylib.c
index 61a461d..6d62c24 100644 (file)
--- a/mylib.c
+++ b/mylib.c
@@ -19,8 +19,7 @@
 
 #include "flatland.h"
 
 
 #include "flatland.h"
 

-int generate_sequence(THByteTensor *output) {
-  long nb_sequences = 1;
+int generate_sequence(long nb_sequences, THByteTensor *output) {

   long nb_images_per_sequence = 5;
   long depth = 3;
   long width = 64;
   long nb_images_per_sequence = 5;
   long depth = 3;
   long width = 64;