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}