github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix nick

Browse source
This commit is contained in:
Colombo 2020-03-07 19:20:07 +04:00
parent d0c280a902
commit d6a685887f
3 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
core/leras/archis.py
View file

@ -207,10 +207,10 @@ def initialize_archis(nn):
nn.get_ae_models = get_ae_models nn.get_ae_models = get_ae_models
def get_ae_models_chervoniy(resolution): def get_ae_models_chervonij(resolution):
lowest_dense_res = resolution // 32 lowest_dense_res = resolution // 32
""" """
by @chervoniy by @chervonij
""" """
conv_kernel_initializer = nn.initializers.ca() conv_kernel_initializer = nn.initializers.ca()
@ -364,7 +364,7 @@ def initialize_archis(nn):
return lowest_dense_res, Encoder, Inter, Decoder return lowest_dense_res, Encoder, Inter, Decoder
nn.get_ae_models_chervoniy = get_ae_models_chervoniy nn.get_ae_models_chervonij = get_ae_models_chervonij
""" """
def get_ae_models2(): def get_ae_models2():

2
core/leras/nn.py
View file

@ -90,7 +90,7 @@ class nn():
# Arhis # Arhis
get_ae_models = None get_ae_models = None
get_ae_models_chervoniy = None get_ae_models_chervonij = None
@staticmethod @staticmethod
def initialize(device_config=None, floatx="float32", data_format="NHWC"): def initialize(device_config=None, floatx="float32", data_format="NHWC"):

2
models/Model_SAEHD/Model.py
View file

@ -177,7 +177,7 @@ class SAEHDModel(ModelBase):
# Initializing model classes # Initializing model classes
if archi == 'liaech': if archi == 'liaech':
lowest_dense_res, Encoder, Inter, Decoder = nn.get_ae_models_chervoniy(resolution) lowest_dense_res, Encoder, Inter, Decoder = nn.get_ae_models_chervonij(resolution)
else: else:
lowest_dense_res, Encoder, Inter, Decoder = nn.get_ae_models(resolution) lowest_dense_res, Encoder, Inter, Decoder = nn.get_ae_models(resolution)