X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland-test.py;h=8d9f11ba02f4af801d627ed8b11d53c20fbd9abd;hp=529284544d40b7bb23c008b09f6b7201e17f1bf6;hb=c99cbb5f30e1e2052bfc36880a1425db600454b5;hpb=d0756dc8a3e42aae30d1c7f543931a2316981635 diff --git a/flatland-test.py b/flatland-test.py index 5292845..8d9f11b 100755 --- a/flatland-test.py +++ b/flatland-test.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python-for-pytorch +#!/usr/bin/env python # # flatland is a simple 2d physical simulator @@ -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',