otx.algorithms.visual_prompting.adapters.pytorch_lightning.models.utils.mlp_block#
MLP block module.
Classes
|
MLPBlock module. |
- class otx.algorithms.visual_prompting.adapters.pytorch_lightning.models.utils.mlp_block.MLPBlock(embedding_dim: int, mlp_dim: int, act: ~typing.Type[~torch.nn.modules.module.Module] = <class 'torch.nn.modules.activation.GELU'>)[source]#
Bases:
Module
MLPBlock module.
Reference: facebookresearch/segment-anything
- Parameters:
Initializes internal Module state, shared by both nn.Module and ScriptModule.