From 018c86c959214a431b9022f40e9c38c51b8f8626 Mon Sep 17 00:00:00 2001
From: deeppomf <deeppomf@gmail.com>
Date: Tue, 27 Feb 2018 23:15:57 -0500
Subject: [PATCH] added comment about batch_size

---
 decensor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/decensor.py b/decensor.py
index 48f5c65..a33ee97 100644
--- a/decensor.py
+++ b/decensor.py
@@ -11,6 +11,7 @@ from model import Model
 from poisson_blend import blend
 from config import *
 
+#TODO: allow variable batch sizes when decensoring. changing BATCH_SIZE will likely result in crashing
 BATCH_SIZE = 1
 
 mask_color = [0, 255, 0]
@@ -81,5 +82,4 @@ def get_mask(x_batch):
 if __name__ == '__main__':
     if not os.path.exists(args.decensor_output_path):
         os.makedirs(args.decensor_output_path)
-    decensor(args)
-    
+    decensor(args)
\ No newline at end of file