min vram -0.5gb for all models

This commit is contained in:
iperov committed 2018-06-05 16:23:43 +04:00
1 parent d5f5a35945
commit 02331c301b
5 files changed
+5 -5

No files matched your search

+1 -1
View File
@@ -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