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

Replaces Convolution operations with MaxPool #21

Merged
merged 2 commits into from
Dec 17, 2018
Merged

Replaces Convolution operations with MaxPool #21

merged 2 commits into from
Dec 17, 2018

Conversation

miguelvr
Copy link
Owner

Fixes #16
Fixes #20

Description:

Substantially simplifies the code by replacing F.conv2d and F.conv3d operations by F.maxpool2d and F.maxpool3d.

Should also result in a performance increase, due to the lightweight nature of the maxpool operation.

Also added unit tests to check against CUDA operations

@miguelvr miguelvr merged commit 8ca5d15 into master Dec 17, 2018
@miguelvr miguelvr deleted the maxpool branch December 17, 2018 11:11
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

Successfully merging this pull request may close these issues.

1 participant