mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-15 01:23:44 -07:00
7 lines
257 B
Python
7 lines
257 B
Python
from .ModelBase import ModelBase
|
|
|
|
def import_model(model_class_name, use_bn=False):
|
|
module = __import__('Model_'+model_class_name, globals(), locals(), [], 1)
|
|
if use_bn:
|
|
return getattr(module, 'Model_bn')
|
|
return getattr(module, 'Model')
|