CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix ModelBase, nnlib

Browse Source
This commit is contained in:
iperov
2019-03-13 20:53:59 +04:00
parent da0fc2d2e1
commit 8da47fec13
3 changed files with 74 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/ModelBase.py
+1 -1
View File
@@ -90,7 +90,7 @@ class ModelBase(object):
if self.iter == 0 or ask_override:
default_batch_size = 0 if self.iter == 0 else self.options.get('batch_size',0)
self.options['batch_size'] = max(0, io.input_int("Batch_size (?:help skip:0/default) : ", default_batch_size, help_message="Larger batch size is always better for NN's generalization, but it can cause Out of Memory error. Tune this value for your videocard manually."))
self.options['batch_size'] = max(0, io.input_int("Batch_size (?:help skip:%d) : " % (default_batch_size), default_batch_size, help_message="Larger batch size is always better for NN's generalization, but it can cause Out of Memory error. Tune this value for your videocard manually."))
else:
self.options['batch_size'] = self.options.get('batch_size', 0)