File size: 522 Bytes
0d8aa44 acb2ce3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | """Export reusable Cross-Shuffle, MIL pooling and composite bag-processing components."""
from .cross_shuffle import CrossTokenShuffle
from .mil_head import MILHead as MILHeadLegacy
from .mil_head_beta import MILHead
from .mil_core import (
CrossShuffleStage,
CrossShuffleTransMIL,
MILCore,
MILHeadStage,
TransStage,
)
__all__ = [
"CrossTokenShuffle",
"MILHead",
"MILHeadLegacy",
"MILCore",
"CrossShuffleStage",
"TransStage",
"MILHeadStage",
"CrossShuffleTransMIL",
]
|