X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=9f825941bbba6d6cf00a4ea72cbbaef008cab7be;hb=128d372813e99d8474bb6e967d5c7e7f085c819d;hp=69731ff89e5e290b2124d84eeffad5aafcabef03;hpb=ac3d9ba45d72a7f3e399de4e3614698ac5e0ce39;p=picoclvr.git

diff --git a/main.py b/main.py
index 69731ff..9f82594 100755
--- a/main.py
+++ b/main.py
@@ -104,6 +104,8 @@ parser.add_argument("--rpl_no_prog", action="store_true", default=False)
 
 parser.add_argument("--grid_size", type=int, default=6)
 
+parser.add_argument("--grid_fraction_play", type=float, default=0)
+
 ##############################
 # picoclvr options
 
@@ -554,6 +556,7 @@ elif args.task == "grid":
         nb_test_samples=args.nb_test_samples,
         batch_size=args.batch_size,
         size=args.grid_size,
+        fraction_play=args.grid_fraction_play,
         logger=log_string,
         device=device,
     )