From d5f5a3594583504a38257390fdf7631169cb8750 Mon Sep 17 00:00:00 2001 From: iperov Date: Tue, 5 Jun 2018 16:20:40 +0400 Subject: [PATCH] min 1.5gb for H64 --- models/Model_H64/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_H64/Model.py b/models/Model_H64/Model.py index 611532f..9e775c3 100644 --- a/models/Model_H64/Model.py +++ b/models/Model_H64/Model.py @@ -15,7 +15,7 @@ class Model(ModelBase): #override def onInitialize(self, **in_options): - self.set_vram_batch_requirements( {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:4,4:8,5:16,6:32,7:32,8:32,9:48} ) ae_input_layer = self.keras.layers.Input(shape=(64, 64, 3)) mask_layer = self.keras.layers.Input(shape=(64, 64, 1)) #same as output