X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=tasks.py;h=ecf0a652185f5733a1bb175b64db8e61b89b0276;hb=fdce96c3960f5f544bd68c0f18cc5cd096ecbfb3;hp=125432361bb4899eb3a64479339c130622f0c5ed;hpb=cdf7c042612ee3b7e5cc2277e82399693a3bf343;p=culture.git diff --git a/tasks.py b/tasks.py index 1254323..ecf0a65 100755 --- 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