-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
Dilation for convolutional layers #766
Conversation
Added tests for |
Marked as ready for review, still need to add documentation though. |
Not sure how the change in the docs works. @awni would you mind helping me here? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, thanks! The docs will be updated the next time we build them.
* add dilation parameter to Conv1d layer * space here too * add conv1d dilation test * add dilation parameter for Conv2d layer * conv2d dilation test
Dilation for convolutional layers
Adding
dilation
parameter tonn.Conv1d
andnn.Conv2d
layers.Reference: #765
Checklist
pre-commit run --all-files
to format my code / installed pre-commit prior to committing changes