Spaces:

Balaji23
/

Meta-Tryon

Runtime error

Meta-Tryon / preprocess /humanparsing /networks /__init__.py

Upload 988 files

0ae9781 verified over 2 years ago

321 Bytes

from __future__ import absolute_import
from networks.AugmentCE2P import resnet101

__factory = {
    'resnet101': resnet101,
}


def init_model(name, *args, **kwargs):
    if name not in __factory.keys():
        raise KeyError("Unknown model arch: {}".format(name))
    return __factory[name](*args, **kwargs)

	from __future__ import absolute_import
	from networks.AugmentCE2P import resnet101

	__factory = {
	'resnet101': resnet101,
	}


	def init_model(name, args, *kwargs):
	if name not in __factory.keys():
	raise KeyError("Unknown model arch: {}".format(name))
	return __factory[name](args, *kwargs)