imdb-transformer
imdb-transformer copied to clipboard
b.text[0] ?
Hi @jensjepsen thanks for the code!
I have question about this line of code:
model_out = model(b.text[0].to(device))
Why does here the input b.text[0]
? Doesn't it mean you only feed the first word?
Please correct me if I am wrong. Thanks.