TextAugmentation-GPT2

TextAugmentation-GPT2 copied to clipboard

generate.py cur_ids = torch.cat([cur_ids, torch.ones((1,1)).long().to(device) * next_token_id], dim = 1) name 'device' is not defined

4

NameError: name 'device' is not defined ?How can I solve it, thank you

xsyzka

Facing issue in generate.py

1

--------------------------------------------------------------------------- RuntimeError Traceback (most recent call last) in 93 TOKENIZER, MODEL = load_models(MODEL_NAME) 94 ---> 95 generate(TOKENIZER, MODEL, SENTENCES, LABEL, DEVICE) in generate(tokenizer, model, sentences, label, device) 47 48 next_token_id...

SarangShaikh201

请问作者方便把数据集发给我吗。麻烦啦

wangcuilin

Thank you for your sharing.I've been studying your code. Now I want it use on multi-label,but I found your code and examples is on a single label. If I want...

haoc111

sayantanbasu05

I wanna predict sentences without calling generate function multiple times.

theneuronprogrammer