diff --git a/models/Model_DF/Model.py b/models/Model_DF/Model.py index ce907a2..c791503 100644 --- a/models/Model_DF/Model.py +++ b/models/Model_DF/Model.py @@ -16,7 +16,7 @@ class Model(ModelBase): #override def onInitialize(self, **in_options): - self.set_vram_batch_requirements( {5:16,6:16,7:16,8:24,9:24,10:32,11:32,12:32,13:48} ) + self.set_vram_batch_requirements( {4.5:16,5:16,6:16,7:16,8:24,9:24,10:32,11:32,12:32,13:48} ) ae_input_layer = self.keras.layers.Input(shape=(128, 128, 3)) mask_layer = self.keras.layers.Input(shape=(128, 128, 1)) #same as output diff --git a/models/Model_H128/Model.py b/models/Model_H128/Model.py index 1b6ab97..5b5cdd6 100644 --- a/models/Model_H128/Model.py +++ b/models/Model_H128/Model.py @@ -17,7 +17,7 @@ class Model(ModelBase): #override def onInitialize(self, **in_options): - self.set_vram_batch_requirements( {3:2,4:2,4:4,5:8,6:8,7:16,8:16,9:24,10:24,11:32,12:32,13:48} ) + self.set_vram_batch_requirements( {2.5:2,3:2,4:2,4:4,5:8,6:8,7:16,8:16,9:24,10:24,11:32,12:32,13:48} ) ae_input_layer = self.keras.layers.Input(shape=(128, 128, 3)) mask_layer = self.keras.layers.Input(shape=(128, 128, 1)) #same as output diff --git a/models/Model_LIAEF128/Model.py b/models/Model_LIAEF128/Model.py index df5105c..7963512 100644 --- a/models/Model_LIAEF128/Model.py +++ b/models/Model_LIAEF128/Model.py @@ -17,7 +17,7 @@ class Model(ModelBase): #override def onInitialize(self, **in_options): - self.set_vram_batch_requirements( {5:4,6:8,7:12,8:16,9:20,10:24,11:24,12:32,13:48} ) + self.set_vram_batch_requirements( {4.5:4,5:4,6:8,7:12,8:16,9:20,10:24,11:24,12:32,13:48} ) ae_input_layer = self.keras.layers.Input(shape=(128, 128, 3)) mask_layer = self.keras.layers.Input(shape=(128, 128, 1)) #same as output diff --git a/models/Model_LIAEF128YAW/Model.py b/models/Model_LIAEF128YAW/Model.py index aa7f668..87e83f4 100644 --- a/models/Model_LIAEF128YAW/Model.py +++ b/models/Model_LIAEF128YAW/Model.py @@ -17,7 +17,7 @@ class Model(ModelBase): #override def onInitialize(self, **in_options): - self.set_vram_batch_requirements( {5:4,6:8,7:12,8:16,9:20,10:24,11:24,12:32,13:48} ) + self.set_vram_batch_requirements( {4.5:4,5:4,6:8,7:12,8:16,9:20,10:24,11:24,12:32,13:48} ) ae_input_layer = self.keras.layers.Input(shape=(128, 128, 3)) mask_layer = self.keras.layers.Input(shape=(128, 128, 1)) #same as output diff --git a/models/Model_MIAEF128/Model.py b/models/Model_MIAEF128/Model.py index 9c09fe0..160253d 100644 --- a/models/Model_MIAEF128/Model.py +++ b/models/Model_MIAEF128/Model.py @@ -21,7 +21,7 @@ class Model(ModelBase): #override def onInitialize(self, **in_options): - self.set_vram_batch_requirements( {5:4,6:8,7:12,8:16,9:20,10:24,11:24,12:32,13:48} ) + self.set_vram_batch_requirements( {4.5:4,5:4,6:8,7:12,8:16,9:20,10:24,11:24,12:32,13:48} ) ae_input_layer = self.keras.layers.Input(shape=(128, 128, 3)) mask_layer = self.keras.layers.Input(shape=(128, 128, 1)) #same as output