segformer_plusplus/__init__.py

from .build_model import create_model, create_custom_model
from .random_benchmark import random_benchmark

__all__ = ['create_model', 'create_custom_model', 'random_benchmark']