SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Reformer] Move model card to google model (#4113)

Browse Source 
* correct model card

* remove model card from patrick von platen
This commit is contained in:
Patrick von Platen
2020-05-02 10:25:22 +02:00
committed by GitHub
parent d713cfc5eb
commit 2d8340a91f
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
model_cards/patrickvonplaten/reformer-crime-and-punish/README.md → model_cards/google/reformer-crime-and-punishment/README.md
View File

@@ -3,9 +3,8 @@ Crime and Punishment text was taken from `gs://trax-ml/reformer/crime-and-punish
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 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
Weights were converted to Hugging Face PyTorch `ReformerModelWithLMHead`. Weights were converted to Hugging Face PyTorch `ReformerModelWithLMHead`.
Model is used as a proof of concept that the forward pass works for a `ReformerModelWithLMHead`. The model is a character level language model. Language from in `crime-and-punishment` style
Given that the model was trained only for 30mins on a ~0.5M tokens dataset and has only 320 tokens, can be generated as follows:
the generation results are reasonable:
```python ```python
model = ReformerModelWithLMHead.from_pretrained("patrickvonplaten/reformer-crime-and-punish") model = ReformerModelWithLMHead.from_pretrained("patrickvonplaten/reformer-crime-and-punish")