Update.

[culture.git] / tasks.py

diff --git a/tasks.py b/tasks.py
index 1254323..ecf0a65 100755 (executable)
--- a/tasks.py
+++ b/tasks.py
@@ -101,8 +101,8 @@ class World(Task):
 
         self.batch_size = batch_size
         self.device = device
-        self.height = 7
-        self.width = 9
+        self.height = 6
+        self.width = 8
 
         self.train_input = world.generate_seq(
             nb_train_samples, height=self.height, width=self.width