lightautoml.ml_algo

Torch utils.

Pooling Strategies

SequenceAbstractPooler

Abstract pooling class.

SequenceClsPooler

CLS token pooling.

SequenceMaxPooler

Max value pooling.

SequenceSumPooler

Sum value pooling.

SequenceAvgPooler

Mean value pooling.

SequenceIndentityPooler

Identity pooling.