mmpretrain/mmcls/models/__init__.py

5 lines
99 B
Python

from .builder import build_model
from .registry import MODELS
__all__ = ['build_model', 'MODELS']