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

Add nn.Dropout support #271

Open
kamathhrishi opened this issue Jul 10, 2021 · 0 comments
Open

Add nn.Dropout support #271

kamathhrishi opened this issue Jul 10, 2021 · 0 comments
Labels
enhancement New feature or request feature Add a new functionality

Comments

@kamathhrishi
Copy link
Contributor

Feature Description

Dropout is very important component of neural network which prevents overfitting. We need to add it's support in order to provide capability to train practical neural networks. Adding support could be as simple as sampling random indices of weights and performing a public multiplication with 0. Also we need to have a way to toggle between model.eval() and model.train().

@kamathhrishi kamathhrishi added enhancement New feature or request feature Add a new functionality labels Jul 10, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request feature Add a new functionality
Projects
None yet
Development

No branches or pull requests

1 participant