<reponame>zhiiker/distribuuuu<filename>distribuuuu/models/__init__.py
from distribuuuu.models.botnet import botnet50  # noqa
from distribuuuu.models.densenet import *  # noqa
from distribuuuu.models.resnet import *  # noqa


def build_model(arch, **kwargs):
    return globals()[arch](**kwargs)

from distribuuuu.models.botnet import botnet50  # noqa
from distribuuuu.models.densenet import *  # noqa
from distribuuuu.models.resnet import *  # noqa


def build_model(arch, **kwargs):
    return globals()[arch](**kwargs)
# noqa # noqa # noqa