From 3a44efa77506655c7c93a7e0e984de3326d2cea2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ren=C3=A9=20Knaebel?= <rknaebel@uni-potsdam.de>
Date: Mon, 11 Sep 2017 12:59:13 +0200
Subject: [PATCH] minor changes for pauls test

---
 main.py      | 1 -
 run_model.sh | 4 ++--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index dc85a85..17ab37b 100644
--- a/main.py
+++ b/main.py
@@ -201,7 +201,6 @@ def main_train(param=None):
                   epochs=args.epochs,
                   callbacks=callbacks,
                   shuffle=True,
-                  validation_split=0.3,
                   class_weight=custom_class_weights)
 
     else:
diff --git a/run_model.sh b/run_model.sh
index 0eb5477..45fce6b 100644
--- a/run_model.sh
+++ b/run_model.sh
@@ -21,8 +21,8 @@ do
                 --filter_embd 256 --kernel_embd 8 --dense_embd 128 \
                 --domain_embd 32 \
                 --filter_main 32  --kernel_main 8 --dense_main 1024 \
-                --batch 256 \
-                --balanced_weights \
+                --batch 128 \
+#                --balanced_weights \
                 --model_output ${OUTPUT} \
                 --type ${TYPE} \
                 --depth ${DEPTH}