fixes

durablenapkin · Jun 28, 2018 · 8bd0e73 · 8bd0e73
1 parent 37978aa
commit 8bd0e73
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/constants.py b/constants.py
@@ -1,6 +1,6 @@
 import os
 
-BATCH_SIZE = os.getenv('BATCHSIZE', 8)
+BATCH_SIZE = os.getenv('BATCH_SIZE', 8)
 MAX_STRING_LEN = 255
 MODEL = os.getenv('MODEL', 'model.h5')
 WEIGHTS = os.getenv('WEIGHTS', 'weights.h5')
diff --git a/train.py b/train.py
@@ -2,7 +2,7 @@
 
 import os
 
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
+#os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 import warnings
 warnings.simplefilter(action='ignore', category=FutureWarning)
 
@@ -17,6 +17,9 @@
 
 from constants import MAX_STRING_LEN, MODEL, WEIGHTS, BATCH_SIZE
 
+BATCH_SIZE = int(BATCH_SIZE)
+MAX_STRING_LEN = int(MAX_STRING_LEN)
+
 NEURONS = os.getenv('TF_NEURONS', 128)  # g00gle.com 256?
 EMBEDDED_DIM = 300