+parser = argparse.ArgumentParser(
+ description = 'Simple convnet test on the SVRT.',
+ formatter_class = argparse.ArgumentDefaultsHelpFormatter
+)
+
+parser.add_argument('--nb_train_samples',
+ type = int, default = 100000,
+ help = 'How many samples for train')
+
+parser.add_argument('--nb_test_samples',
+ type = int, default = 10000,
+ help = 'How many samples for test')
+
+parser.add_argument('--nb_epochs',
+ type = int, default = 25,
+ help = 'How many training epochs')
+
+parser.add_argument('--log_file',
+ type = str, default = 'cnn-svrt.log',
+ help = 'Log file name')
+
+args = parser.parse_args()
+
+######################################################################
+
+log_file = open(args.log_file, 'w')
+
+print(Fore.RED + 'Logging into ' + args.log_file + Style.RESET_ALL)
+
+def log_string(s):
+ s = Fore.GREEN + time.ctime() + Style.RESET_ALL + ' ' + \
+ str(problem_number) + ' ' + s
+ log_file.write(s + '\n')
+ log_file.flush()
+ print(s)
+
+######################################################################
+