Skip to content

Commit a58f916

Browse files
committed
Missed __init__.py update for attention pooling layer add
1 parent 59b6222 commit a58f916

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

timm/layers/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from .activations import *
22
from .adaptive_avgmax_pool import \
33
adaptive_avgmax_pool2d, select_adaptive_pool2d, AdaptiveAvgMaxPool2d, SelectAdaptivePool2d
4+
from .attention_pool import AttentionPoolLatent
45
from .attention_pool2d import AttentionPool2d, RotAttentionPool2d, RotaryEmbedding
56
from .blur_pool import BlurPool2d
67
from .classifier import ClassifierHead, create_classifier, NormMlpClassifierHead

0 commit comments

Comments
 (0)