From adc1e701de664cf28feab5c9ca4e10f63ae66904 Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 7 Jun 2018 10:06:24 +0400 Subject: [PATCH] increased H64 decoder size for 4GB+ --- models/Model_H64/Model.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/models/Model_H64/Model.py b/models/Model_H64/Model.py index 2b6293c..e4ef157 100644 --- a/models/Model_H64/Model.py +++ b/models/Model_H64/Model.py @@ -18,7 +18,7 @@ class Model(ModelBase): tf = self.tf keras = self.keras K = keras.backend - self.set_vram_batch_requirements( {1.5:2,2:2,3:4,4:8,5:16,6:32,7:32,8:32,9:48} ) + self.set_vram_batch_requirements( {1.5:2,2:2,3:8,4:16,5:24,6:32,7:40,8:48} ) bgr_shape, mask_shape, self.encoder, self.decoder_src, self.decoder_dst = self.Build(self.created_vram_gb) if not self.is_first_run(): @@ -157,14 +157,19 @@ class Model(ModelBase): def Decoder(): if created_vram_gb >= 4: input_ = self.keras.layers.Input(shape=(8, 8, 512)) + x = input_ + x = upscale(self.keras, x, 512) + x = upscale(self.keras, x, 256) + x = upscale(self.keras, x, 128) + else: input_ = self.keras.layers.Input(shape=(8, 8, 256)) - x = input_ - x = upscale(self.keras, x, 256) - x = upscale(self.keras, x, 128) - x = upscale(self.keras, x, 64) - + x = input_ + x = upscale(self.keras, x, 256) + x = upscale(self.keras, x, 128) + x = upscale(self.keras, x, 64) + y = input_ #mask decoder y = upscale(self.keras, y, 256) y = upscale(self.keras, y, 128)