github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-20 05:23:22 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Propagated BN implementation, no CLI support yet.

Browse source
This commit is contained in:
Jose 2023-02-07 13:19:45 +01:00 • committed by GitHub
commit 7439d5003e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 1133 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/__init__.py
View file

@ -1,5 +1,7 @@
from .ModelBase import ModelBase
def import_model(model_class_name):
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')