search

graykode / gpt-2-Pytorch

Simple Text-Generator with OpenAI gpt-2 Pytorch Implementation
MIT License
963 stars 225 forks source link

When --batch_size bigger than 1, error occurs #16

Open eecrazy opened 4 years ago

eecrazy commented 4 years ago

python main.py --text "Babies cry because" --length 25 --nsamples 5 --batch_size 5