jsksxs360/How-to-use-Transformers

end_token_index = torch.where(labels == tokenizer.eos_token_id)[1]

harlen1212 opened this issue · 0 comments

harlen1212 commented
end_token_index = torch.where(labels == tokenizer.eos_token_id)[1]