Todo Generate text every few epochs and log the text to tensorflow (with greedy and beam search) Handle arguments to init() method of a model in checkpointing add command line option for "gru" or "lstm"