refactor, move pointwise_avg_3d into pointwise_avg

Signed-off-by: chenhe <chenhe@megarobo.tech>
QUVA-Lab · Jul 18, 2023 · 388af7f · 388af7f
1 parent 14e0c09
commit 388af7f
Show file tree

Hide file tree

Showing 3 changed files with 233 additions and 364 deletions.
diff --git a/escnn/nn/modules/pooling/__init__.py b/escnn/nn/modules/pooling/__init__.py
@@ -4,16 +4,13 @@
 from .pointwise_max import PointwiseMaxPoolAntialiased2D, PointwiseMaxPoolAntialiased
 from .pointwise_max import PointwiseMaxPool3D
 from .pointwise_max import PointwiseMaxPoolAntialiased3D
-from .pointwise_avg import PointwiseAvgPool, PointwiseAvgPool2D
-from .pointwise_avg import PointwiseAvgPoolAntialiased, PointwiseAvgPoolAntialiased2D
+from .pointwise_avg import PointwiseAvgPool, PointwiseAvgPool2D, PointwiseAvgPool3D
+from .pointwise_avg import PointwiseAvgPoolAntialiased, PointwiseAvgPoolAntialiased2D, PointwiseAvgPoolAntialiased3D
 from .pointwise_adaptive_avg import PointwiseAdaptiveAvgPool2D, PointwiseAdaptiveAvgPool
 from .pointwise_adaptive_avg import PointwiseAdaptiveAvgPool3D
 from .pointwise_adaptive_max import PointwiseAdaptiveMaxPool2D, PointwiseAdaptiveMaxPool
 from .pointwise_adaptive_max import PointwiseAdaptiveMaxPool3D
 
-from .pointwise_avg_3d import PointwiseAvgPool3D
-from .pointwise_avg_3d import PointwiseAvgPoolAntialiased3D
-
 
 __all__ = [
     "NormMaxPool",
@@ -31,5 +28,3 @@
     "PointwiseAvgPool3D",
     "PointwiseAvgPoolAntialiased3D",
 ]
-
-