+if args.dirty_debug:
+ args.nb_train_samples = 2500
+ args.nb_test_samples = 100
+
+if args.physical_batch_size is None:
+ args.physical_batch_size = args.batch_size
+else:
+ assert args.batch_size % args.physical_batch_size == 0
+
+assert args.nb_train_samples % args.batch_size == 0
+assert args.nb_test_samples % args.batch_size == 0
+
+quizz_machine = quizz_machine.QuizzMachine(
+ problem=sky.Sky(height=6, width=8, nb_birds=3, nb_iterations=2, speed=2),
+ nb_train_samples=args.nb_train_samples,
+ nb_test_samples=args.nb_test_samples,
+ batch_size=args.physical_batch_size,
+ result_dir=args.result_dir,
+ logger=log_string,
+ device=device,
+)