-parser.add_argument("--learning_rate_schedule", type=str, default="10: 2e-5,30: 4e-6")
-
-parser.add_argument("--dim_model", type=int, default=512)
-
-parser.add_argument("--dim_keys", type=int, default=64)
-
-parser.add_argument("--dim_hidden", type=int, default=2048)
-
-parser.add_argument("--nb_heads", type=int, default=8)
-
-parser.add_argument("--nb_blocks", type=int, default=12)
-
-parser.add_argument("--dropout", type=float, default=0.1)
-
-parser.add_argument("--deterministic_synthesis", action="store_true", default=False)
-
-parser.add_argument("--no_checkpoint", action="store_true", default=False)
-
-parser.add_argument("--overwrite_results", action="store_true", default=False)
-
-parser.add_argument("--checkpoint_name", type=str, default="checkpoint.pth")
-
-##############################
-# picoclvr options
-
-parser.add_argument("--picoclvr_nb_colors", type=int, default=5)
-
-parser.add_argument("--picoclvr_height", type=int, default=12)
-
-parser.add_argument("--picoclvr_width", type=int, default=16)
-
-parser.add_argument("--picocvlr_prune_properties", type=str, default="none")
-
-##############################
-# Maze options
-
-parser.add_argument("--maze_height", type=int, default=23)
-
-parser.add_argument("--maze_width", type=int, default=39)
-
-parser.add_argument("--maze_nb_walls", type=int, default=45)