added depth parameter in config.py

n0obcoder · web-flow · commit 4cd75602edb0 · 2019-11-08T01:28:13.000+05:30
diff --git a/config.py b/config.py
@@ -14,6 +14,9 @@
 noisy_dir = 'noisy'
 debug_dir = 'debug'
 
+# depth of UNet 
+depth = 5 # try decreasing the depth value if there is a memory error
+
 # text file to get text from
 txt_file_dir = 'shitty_text.txt'
 
