def _main():
# Set random seed for training
seed = 8964
tf.set_random_seed(seed)
np.random.seed(seed)
random.seed(seed)
pp = pprint.PrettyPrinter()
print_flags_dict = {}
for key in FLAGS.__flags.keys():
print_flags_dict[key] = getattr(FLAGS, key)
pp.pprint(print_flags_dict)
if not os.path.exists(FLAGS.checkpoint_dir):
os.makedirs(FLAGS.checkpoint_dir)
assert os.path.exists(FLAGS.train_dir), "Specify training data directory!"
assert os.path.exists(FLAGS.val_dir), "Specify validation data directory!"
trl = Learner()
trl.train(FLAGS)
