Fine-tuned pre-trained GPT2 for custom topic specific text generation. Such system can be used for Text Augmentation.
187
stars
43
forks
source link
generate.py cur_ids = torch.cat([cur_ids, torch.ones((1,1)).long().to(device) * next_token_id], dim = 1) #11
Closed
xsyzka closed 2 years ago