+import svrt
+
+######################################################################
+
+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')
+
+args = parser.parse_args()
+
+######################################################################
+
+log_file = open('cnn-svrt.log', 'w')
+
+def log_string(s):
+ s = time.ctime() + ' ' + str(problem_number) + ' | ' + s
+ log_file.write(s + '\n')
+ log_file.flush()
+ print(s)