This commit is contained in:
Colombo committed 2020-03-07 19:20:07 +04:00
1 parent d0c280a902
commit d6a685887f
3 files changed
+5 -5

No files matched your search

+1 -1
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)