Update ThunderNet_PSRoIAlignExtractor.py

alpc128 · web-flow · commit e13f604950db · 2022-03-29T00:01:24.000+08:00
diff --git a/thundernet/ThunderNet_PSRoIAlignExtractor.py b/thundernet/ThunderNet_PSRoIAlignExtractor.py
@@ -12,7 +12,7 @@
 from torch.autograd.function import once_differentiable
 import importlib
 
-_C = importlib.import_module('PSROIAlign._C')
+_C = importlib.import_module('thundernet.PSROIAlign._C')
 
 
 class _PSROIAlign(Function):
