argmax(): argument 'input' (position 1) must be Tensor, not str
#1
by
peterphan
- opened
Hi author,
I run the source code in Model in Action and I received this error as in the screenshot above:
error message: 'argmax(): argument 'input' (position 1) must be Tensor, not str'
When I print the value of start_scores and end_scores, they are 'start_logits' and 'end_logits'
Could you please help me to fix them?
Thank you so much