Skip to content

Commit de7d9dc

Browse files
authored
Add pooling deprecated (#7)
1 parent f51b952 commit de7d9dc

File tree

2 files changed

+33
-2
lines changed

2 files changed

+33
-2
lines changed

examples/model_zoo/resnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
from tensorlayerx import logging
1616
from tensorlayerx.files import (maybe_download_and_extract)
17-
from tensorlayerx.nn import (BatchNorm, Conv2d, Linear, Elementwise, GlobalMeanPool2d, MaxPool2d)
17+
from tensorlayerx.nn import (BatchNorm, Conv2d, Linear, Elementwise, GlobalAvgPool2d, MaxPool2d)
1818
from tensorlayerx.nn import Module, Sequential
1919

2020
__all__ = [
@@ -159,7 +159,7 @@ def make_layer(self):
159159
else:
160160
layer_list.append(identity_block(3, block_filters[stage - 2], stage=stage, block=block))
161161
elif block_name == 'avg_pool':
162-
layer_list.append(GlobalMeanPool2d(name='avg_pool'))
162+
layer_list.append(GlobalAvgPool2d(name='avg_pool'))
163163
elif block_name == 'fc1000':
164164
layer_list.append(Linear(self.n_classes, name='fc1000', in_features=2048))
165165

tensorlayerx/nn/layers/deprecated.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -477,3 +477,34 @@ def DeConv2d(*args, **kwargs):
477477

478478
def DeConv3d(*args, **kwargs):
479479
raise NonExistingLayerError("DeConv3d --> ConvTranspose3d" + __log__)
480+
481+
482+
def MeanPool1d(*args, **kwargs):
483+
raise NonExistingLayerError("MeanPool1d --> AvgPool1d" + __log__)
484+
485+
def MeanPool2d(*args, **kwargs):
486+
raise NonExistingLayerError("MeanPool2d --> AvgPool2d" + __log__)
487+
488+
def MeanPool3d(*args, **kwargs):
489+
raise NonExistingLayerError("MeanPool3d --> AvgPool3d" + __log__)
490+
491+
def GlobalMeanPool1d(*args, **kwargs):
492+
raise NonExistingLayerError("GlobalMeanPool1d --> GlobalAvgPool1d" + __log__)
493+
494+
def GlobalMeanPool2d(*args, **kwargs):
495+
raise NonExistingLayerError("GlobalMeanPool2d --> GlobalAvgPool2d" + __log__)
496+
497+
def GlobalMeanPool3d(*args, **kwargs):
498+
raise NonExistingLayerError("GlobalMeanPool3d --> GlobalAvgPool3d" + __log__)
499+
500+
def AdaptiveMeanPool1d(*args, **kwargs):
501+
raise NonExistingLayerError("AdaptiveMeanPool1d --> AdaptiveAvgPool1d" + __log__)
502+
503+
def AdaptiveMeanPool2d(*args, **kwargs):
504+
raise NonExistingLayerError("AdaptiveMeanPool2d --> AdaptiveAvgPool2d" + __log__)
505+
506+
def AdaptiveMeanPool3d(*args, **kwargs):
507+
raise NonExistingLayerError("AdaptiveMeanPool3d --> AdaptiveAvgPool3d" + __log__)
508+
509+
__all__ += ['MeanPool1d', 'MeanPool2d', 'MeanPool3d', 'GlobalMeanPool1d', 'GlobalMeanPool2d', 'GlobalMeanPool3d',
510+
'AdaptiveMeanPool1d', 'AdaptiveMeanPool2d', 'AdaptiveMeanPool3d']

0 commit comments

Comments
 (0)