github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 21:12:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

4
models/Model_RecycleGAN/Model.py
View file

@ -36,9 +36,9 @@ class Model(ModelBase):
created_resolution = 64
try:
created_batch_size = int ( input ("Batch_size (minimum/default - 16) : ") )
created_batch_size = int ( input ("Batch_size (minimum/default - 10) : ") )
except:
created_batch_size = 16
created_batch_size = 10
created_batch_size = max(created_batch_size,1)
print ("Done. If training won't start, decrease resolution")