david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

min 1.5gb for H64

Browse Source
This commit is contained in:
iperov
2018-06-05 16:20:40 +04:00
parent 9f48b2c87a
commit d5f5a35945
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_H64/Model.py
+1 -1
View File
@@ -15,7 +15,7 @@ class Model(ModelBase):
#override #override
def onInitialize(self, **in_options): 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)) ae_input_layer = self.keras.layers.Input(shape=(64, 64, 3))
mask_layer = self.keras.layers.Input(shape=(64, 64, 1)) #same as output mask_layer = self.keras.layers.Input(shape=(64, 64, 1)) #same as output