fix arguments for initializing a model

huggingface / pytorch-openai-transformer-lm

🐥A PyTorch implementation of OpenAI's finetuned transformer language model with a script to import the weights pre-trained by OpenAI

MIT License

1.51k stars 285 forks source link

Closed soskek closed 6 years ago

soskek commented 6 years ago

In the sample ROCStory script, the line for model construction seems wrong. This will fix it.

thomwolf commented 6 years ago

Indeed, thanks!