Multi box loss #6946
Closed
Multi box loss #6946
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
wanghaox
force-pushed
the
multi_box_loss
branch
from
1c43d95
to
b37e2c4
Compare
wanghaox
force-pushed
the
multi_box_loss
branch
from
b37e2c4
to
3845be5
Compare
… multi_box_loss
wanghaox
force-pushed
the
multi_box_loss
branch
from
9f4af0c
to
1972fab
Compare
sweetsky0901
reviewed
Jan 3, 2018
| set_constant(cpu_ctx, &d_loc_cpu, 0); | ||
| set_constant(cpu_ctx, &d_conf_cpu, 0); | ||
| loc_offset += | ||
| math::TransposeFromNHWCToNCHW<platform::CPUDeviceContext, T>( |
There was a problem hiding this comment.
TransposeFromNHWCToNCHW好像是可以用GPU的。
用GPU transpose之后再copyfrom到cpu进行接下来的计算是不是更合理一些
There was a problem hiding this comment.
这里原始实现有些晦涩，主要考虑到data layout，如果用tensor表示，而不是matrix，可能要简单一些
|
Since you haven't replied for a long time, we have closed this issue/pr. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
resolve #6818