mirror of
https://github.com/deepinsight/insightface.git
synced 2026-05-14 12:17:55 +00:00
15 lines
445 B
Python
Executable File
15 lines
445 B
Python
Executable File
from mmcv.utils import Registry, build_from_cfg
|
|
|
|
TRANSFORMER = Registry('Transformer')
|
|
POSITIONAL_ENCODING = Registry('Position encoding')
|
|
|
|
|
|
def build_transformer(cfg, default_args=None):
|
|
"""Builder for Transformer."""
|
|
return build_from_cfg(cfg, TRANSFORMER, default_args)
|
|
|
|
|
|
def build_positional_encoding(cfg, default_args=None):
|
|
"""Builder for Position Encoding."""
|
|
return build_from_cfg(cfg, POSITIONAL_ENCODING, default_args)
|