lumin.nn.models.layers package¶
Submodules¶
lumin.nn.models.layers.activations module¶
-
lumin.nn.models.layers.activations.
lookup_act
(act)[source]¶ Map activation name to class
- Parameters
act (
str
) – string representation of activation function- Return type
Any
- Returns
Class implementing requested activation function
-
class
lumin.nn.models.layers.activations.
Swish
(inplace=False)[source]¶ Bases:
torch.nn.modules.module.Module
Non-trainable Swish activation function https://arxiv.org/abs/1710.05941
- Parameters
inplace – whether to apply activation inplace
- Examples::
>>> swish = Swish()