From 46bda1d6835af605e595b8c6a54a6f0a4790d2ed Mon Sep 17 00:00:00 2001 From: iperov Date: Fri, 28 Dec 2018 23:42:05 +0400 Subject: [PATCH] RecycleGAN : set minimum bs to 10 --- models/Model_RecycleGAN/Model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/Model_RecycleGAN/Model.py b/models/Model_RecycleGAN/Model.py index 8fcb9ef..f0d1856 100644 --- a/models/Model_RecycleGAN/Model.py +++ b/models/Model_RecycleGAN/Model.py @@ -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")