minor changes for pauls test

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:

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}