Skip to content
New issue

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

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

Already on GitHub? # 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
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant