added to https://github.com/r-tensorflow/gpt2/pull/2
added to #2