######################################################################
-parser = argparse.ArgumentParser(description='Toy RNN.')
+parser = argparse.ArgumentParser(description='Toy attention model.')
parser.add_argument('--nb_epochs',
type = int, default = 250)