|
|
|
from annotator.uniformer.mmcv.utils import Registry |
|
|
|
CONV_LAYERS = Registry('conv layer') |
|
NORM_LAYERS = Registry('norm layer') |
|
ACTIVATION_LAYERS = Registry('activation layer') |
|
PADDING_LAYERS = Registry('padding layer') |
|
UPSAMPLE_LAYERS = Registry('upsample layer') |
|
PLUGIN_LAYERS = Registry('plugin layer') |
|
|
|
DROPOUT_LAYERS = Registry('drop out layers') |
|
POSITIONAL_ENCODING = Registry('position encoding') |
|
ATTENTION = Registry('attention') |
|
FEEDFORWARD_NETWORK = Registry('feed-forward Network') |
|
TRANSFORMER_LAYER = Registry('transformerLayer') |
|
TRANSFORMER_LAYER_SEQUENCE = Registry('transformer-layers sequence') |
|
|
