From abb1fa3f374811ea09d0bc3440d820c50735008d Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sat, 2 May 2020 10:32:00 +0200 Subject: [PATCH] Update README.md --- model_cards/google/reformer-crime-and-punishment/README.md | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/model_cards/google/reformer-crime-and-punishment/README.md b/model_cards/google/reformer-crime-and-punishment/README.md index 03d54279..b30ef06d 100644 --- a/model_cards/google/reformer-crime-and-punishment/README.md +++ b/model_cards/google/reformer-crime-and-punishment/README.md @@ -5,8 +5,7 @@ Crime and Punishment is a novel written by Fyodor Dostoevsky and was translated Crime and Punishment training data was taken from `gs://trax-ml/reformer/crime-and-punishment-2554.txt` and contains roughly 0.5M tokens. -The ReformerLM model was trained in flax using colab notebook proposed by authors: https://colab.research.google.com/github/google/trax/blob/master/trax/models/reformer/text_generation.ipynb . -and the weights were converted to Hugging Face's PyTorch ReformerLM model `ReformerModelWithLMHead`. +The ReformerLM model was trained in flax using colab notebook proposed by authors: https://colab.research.google.com/github/google/trax/blob/master/trax/models/reformer/text_generation.ipynb and the weights were converted to Hugging Face's PyTorch ReformerLM model `ReformerModelWithLMHead`. The model is a language model that operates on small sub-word units. Text can be generated as follows: