From 28549dc153948f3dd91bddfdc57ea9cc7cb87d66 Mon Sep 17 00:00:00 2001 From: Colombo Date: Sat, 28 Dec 2019 16:50:33 +0400 Subject: [PATCH] SAEHD:optimized architecture, you have to restart training --- models/Model_SAEHD/Model.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index b02f2de..cc9217c 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -198,8 +198,6 @@ class SAEHDModel(ModelBase): if dims % 2 != 0: dims += 1 - - def func(x): for i in [8,4,2]: @@ -213,7 +211,7 @@ class SAEHDModel(ModelBase): x = Add()([x, x0]) x = LeakyReLU(0.2)(x) - return Conv2D(output_nc, kernel_size=5, padding='same', activation='sigmoid')(x) + return Conv2D(output_nc, kernel_size=1, padding='same', activation='sigmoid')(x) return func @@ -327,7 +325,7 @@ class SAEHDModel(ModelBase): x = Add()([x, x0]) x = LeakyReLU(0.2)(x) - return Conv2D(output_nc, kernel_size=5, padding='same', activation='sigmoid')(x) + return Conv2D(output_nc, kernel_size=1, padding='same', activation='sigmoid')(x) return func