/caffe

Fix #2406: wrong thread blocks setting for PReLU #2414

Merged
merged 2 commits into from May 5, 2015

Conversation

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
2 participants
@tnarihi @jeffdonahue
@tnarihi
Contributor

tnarihi commented May 4, 2015

This PR corrects the argument of CAFFE_GET_BLOCKS for parameter backward computation in PReLULayer pointed in #2406. This doesn't change anything numerically, but will speed up computation by removing redundant blocks. See 63ed23b for primary change in this PR.
Some modification for readability is also made (4348c6f).

Thanks @mausoomsarkar for reporting!

tnarihi referenced this pull request May 4, 2015

Closed

PReLU is slow #2406

@jeffdonahue
Contributor

jeffdonahue commented May 5, 2015

Thanks for the fix and cleanup @tnahari!

@jeffdonahue jeffdonahue added a commit that referenced this pull request May 5, 2015

@jeffdonahue jeffdonahue Merge pull request #2414 from tnarihi/fix-prelu-redanduncy 
Fix #2406: wrong thread blocks setting for PReLU
 40f6dff

@jeffdonahue jeffdonahue merged commit 40f6dff into BVLC:master May 5, 2015
1 check passed

1 check passed

continuous-integration/travis-ci/pr The Travis CI build passed
Details
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment