imdb-transformer icon indicating copy to clipboard operation
imdb-transformer copied to clipboard

b.text[0] ?

Open littleflow3r opened this issue 5 years ago • 0 comments

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.

littleflow3r avatar May 23 '19 05:05 littleflow3r