diff --git a/core/leras/archis.py b/core/leras/archis.py index aaaa2dc..6c899f9 100644 --- a/core/leras/archis.py +++ b/core/leras/archis.py @@ -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(): diff --git a/core/leras/nn.py b/core/leras/nn.py index 37a2ed2..be53a8f 100644 --- a/core/leras/nn.py +++ b/core/leras/nn.py @@ -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"): diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 9b647af..3125309 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -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)