Skip to content
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.

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

Declaring classifier_dropout in model config but not using it #12792

Closed
PhilipMay opened this issue Jul 19, 2021 · 5 comments
Closed

Declaring classifier_dropout in model config but not using it #12792

PhilipMay opened this issue Jul 19, 2021 · 5 comments

Comments

@PhilipMay
Copy link
Contributor

The PEGASUS model is declaring classifier_dropout in the model config:

But the model never uses it. This is a bug IMO.

In contrast mBART defines it:

and uses it:

If wanted I can provide a fix (non breaking) as a PR.

@PhilipMay
Copy link
Contributor Author

Tagging @NielsRogge

@PhilipMay
Copy link
Contributor Author

also see this issue: #12781

@PhilipMay
Copy link
Contributor Author

@NielsRogge what do you think?

@PhilipMay
Copy link
Contributor Author

Well, since Pegasus does not have SequenceClassification or TokenClassification the classifier_dropout could be just removed from the config.

@PhilipMay
Copy link
Contributor Author

fixed in PR - closing

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant