X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=17936c34dc7c3ea4a0108a219a3a386151a34755;hb=6e09c88d26d0bfd675af9afd9cdc32aa3485d1b7;hp=496a6034b857d303baa1a32ab8bb48fd68bf84eb;hpb=503298855a80bde0bf856f1a34b532079d3c7ef6;p=picoclvr.git diff --git a/main.py b/main.py index 496a603..17936c3 100755 --- a/main.py +++ b/main.py @@ -160,10 +160,12 @@ parser.add_argument("--expr_result_max", type=int, default=99) parser.add_argument("--expr_input_file", type=str, default=None) ############################## -# Misc +# Mixing parser.add_argument("--mixing_hard", action="store_true", default=False) +parser.add_argument("--mixing_deterministic_start", action="store_true", default=False) + ###################################################################### args = parser.parse_args() @@ -346,6 +348,8 @@ def log_string(s): sys.stdout.flush() +log_string(f"argv {' '.join(sys.argv)}") + for n in vars(args): log_string(f"args.{n} {getattr(args, n)}") @@ -416,7 +420,9 @@ elif args.task == "twotargets": elif args.task == "mixing": task = tasks.SandBox( - problem=problems.ProblemMixing(hard=args.mixing_hard), + problem=problems.ProblemMixing( + hard=args.mixing_hard, random_start=not args.mixing_deterministic_start + ), nb_train_samples=args.nb_train_samples, nb_test_samples=args.nb_test_samples, batch_size=args.batch_size,