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

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)