towhee.models.collaborative_experts.collaborative_expertsΒΆ
Functions
Create CENet model. |
|
Remove nans, which we expect to find at missing indices. |
|
Compute a similarity matrix from sharded vectors. |
|
Compute a similarity matrix from sharded vectors. |
Classes
CE Module |
|
Collaborative Experts Module. |
|
ContextGating Module |
|
G_reason Module |
|
Applies the mish function element-wise: mish(x) = x * tanh(softplus(x)) = x * tanh(ln(1 + exp(x))) SRC: https://github.com/digantamisra98/Mish/blob/master/Mish/Torch/mish.py |
|
ReduceDim Module |
|
RelationModuleMultiScale Module |
|
RelationModuleMultiScale_Cat Module |
|
TemporalAttention Module |