From 0833a38bb900ec88378ad1aa2009db8512aab6bb Mon Sep 17 00:00:00 2001 From: Colombo Date: Fri, 27 Dec 2019 21:06:45 +0400 Subject: [PATCH] Quick96: optimized architecture --- models/Model_Quick96/Model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/models/Model_Quick96/Model.py b/models/Model_Quick96/Model.py index fe3af01..b9a94c3 100644 --- a/models/Model_Quick96/Model.py +++ b/models/Model_Quick96/Model.py @@ -16,7 +16,7 @@ class Quick96Model(ModelBase): super().__init__(*args, **kwargs, ask_enable_autobackup=False, ask_write_preview_history=False, - ask_target_iter=False, + ask_target_iter=True, ask_batch_size=False, ask_random_flip=False) @@ -92,8 +92,8 @@ class Quick96Model(ModelBase): y = self.upscale(d_dims//2)(y) y = self.upscale(d_dims//4)(y) - return Conv2D(3, kernel_size=5, padding='same', activation='tanh')(x), \ - Conv2D(1, kernel_size=5, padding='same', activation='sigmoid')(y) + return Conv2D(3, kernel_size=1, padding='same', activation='tanh')(x), \ + Conv2D(1, kernel_size=1, padding='same', activation='sigmoid')(y) return func