Cosmetics.

diff --git a/README.md b/README.md
index 340eee4..47633c0 100644 (file)
--- a/README.md
+++ b/README.md
@@ -8,5 +8,5 @@ the pytorch framework.
 ```
 make -j -k
 ./built.py
-./flatland-test.py --nb_sequences 5 --nb_shapes 6 --nb_images_per_sequence 20
+./flatland-test.py
 ```

diff --git a/flatland-test.py b/flatland-test.py
index 5292845..8592a4d 100755 (executable)
--- a/flatland-test.py
+++ b/flatland-test.py
@@ -51,7 +51,7 @@ parser.add_argument('--nb_shapes',
                     help = 'Image height')
 
 parser.add_argument('--nb_sequences',
-                    type = int, default = 4,
+                    type = int, default = 8,
                     help = 'How many sequences to generate')
 
 parser.add_argument('--nb_images_per_sequences',