We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 59b6222 commit a58f916Copy full SHA for a58f916
timm/layers/__init__.py
@@ -1,6 +1,7 @@
1
from .activations import *
2
from .adaptive_avgmax_pool import \
3
adaptive_avgmax_pool2d, select_adaptive_pool2d, AdaptiveAvgMaxPool2d, SelectAdaptivePool2d
4
+from .attention_pool import AttentionPoolLatent
5
from .attention_pool2d import AttentionPool2d, RotAttentionPool2d, RotaryEmbedding
6
from .blur_pool import BlurPool2d
7
from .classifier import ClassifierHead, create_classifier, NormMlpClassifierHead
0 commit comments