New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

IMDB Sentiment Classifier not working #1

Open

MichaelRinger opened this issue Feb 28, 2021 · 0 comments

MichaelRinger commented Feb 28, 2021

Hey,

im trying to do my own transformer for imdb sentiment classification, but i am having a really brainhurting problem with the attention weights.

Ive opened a thread for this at the pytorch forum.

https://discuss.pytorch.org/t/multiheadattention-not-working-in-sentiment-classifier/112634

Take a look if you want,

Thanks

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment