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

RecycleGAN : set minimum bs to 10

Browse Source
This commit is contained in:
iperov
2018-12-28 23:42:05 +04:00
parent 474fff248f
commit 46bda1d683
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_RecycleGAN/Model.py
+2 -2
View File
@@ -36,9 +36,9 @@ class Model(ModelBase):
created_resolution = 64 created_resolution = 64
try: try:
created_batch_size = int ( input ("Batch_size (minimum/default - 16) : ") ) created_batch_size = int ( input ("Batch_size (minimum/default - 10) : ") )
except: except:
created_batch_size = 16 created_batch_size = 10
created_batch_size = max(created_batch_size,1) created_batch_size = max(created_batch_size,1)
print ("Done. If training won't start, decrease resolution") print ("Done. If training won't start, decrease resolution")