This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
Revert "fix(mgb/opr): change EQ opr's backward_graph to nullptr instead of InvalidGrad"
This reverts commit
6355337641
. GitOrigin-RevId:
6897f6d45d
tags/v1.0.0-rc1
Megvii Engine Team
Xinran Xu
4 years ago
parent
14e71b551b
commit
30ce3c60bd
2 changed files
with
1 additions
and
4 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+0
-3
src/opr/impl/dnn/roi_align.cpp
+1
-1
src/opr/impl/dnn/roi_pooling.cpp
+ 0
- 3
src/opr/impl/dnn/roi_align.cpp
View File
@@ -42,9 +42,6 @@ SymbolVar ROIAlignForward::make(SymbolVar src, SymbolVar rois,
#ifdef MGB_ENABLE_GRAD
MGB_IMPL_OPR_GRAD(ROIAlignForward) {
if (out_grad[1]) {
return InvalidGrad::make(opr, wrt_idx);
}
if (wrt_idx == 0) {
// wrt src
SymbolVar grad =
+ 1
- 1
src/opr/impl/dnn/roi_pooling.cpp
View File
@@ -86,7 +86,7 @@ size_t ROIPoolingForward::get_workspace_size_bytes(
#ifdef MGB_ENABLE_GRAD
MGB_IMPL_OPR_GRAD(ROIPoolingForward) {
if (
out_grad[1] ||
wrt_idx == 2) {
if (wrt_idx == 2) {
return InvalidGrad::make(opr, wrt_idx);
}
if (wrt_idx == 0) {
Write
Preview
Loading…
Cancel
Save