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
def get_ae_models_chervoniy(resolution):
def get_ae_models_chervonij(resolution):
lowest_dense_res = resolution // 32
"""
by @chervoniy
by @chervonij
"""
conv_kernel_initializer = nn.initializers.ca()
@ -364,7 +364,7 @@ def initialize_archis(nn):
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():

2
core/leras/nn.py
View file

@ -90,7 +90,7 @@ class nn():
# Arhis
get_ae_models = None
get_ae_models_chervoniy = None
get_ae_models_chervonij = None
@staticmethod
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
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:
lowest_dense_res, Encoder, Inter, Decoder = nn.get_ae_models(resolution)