fixing bugs when trans to TF1.0

CharlesShang · Mar 7, 2017 · 27e4a78 · 27e4a78
1 parent cc4f6a3
commit 27e4a78
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/roi_pooling_layer/roi_pooling_op_grad.py b/lib/roi_pooling_layer/roi_pooling_op_grad.py
@@ -1,5 +1,6 @@
 import tensorflow as tf
 from tensorflow.python.framework import ops
+import roi_pooling_op
 
 @ops.RegisterGradient("RoiPool")
 def _roi_pool_grad(op, grad, _):