From a2213adf55446274d4255ce3e3a4137dc89eed6f Mon Sep 17 00:00:00 2001
From: chaklim <chaklim@gmail.com>
Date: Thu, 5 Dec 2019 14:31:29 +0800
Subject: [PATCH] Update avaiable_slots and prediction batch size

- prevent using too much memory
---
 dataPrepScripts/CreateTensor.py | 2 +-
 shared/param.py                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dataPrepScripts/CreateTensor.py b/dataPrepScripts/CreateTensor.py
index 537ebd1..de515c4 100644
--- a/dataPrepScripts/CreateTensor.py
+++ b/dataPrepScripts/CreateTensor.py
@@ -174,7 +174,7 @@ def samtools_view_process_from(
 
 
 def OutputAlnTensor(args):
-    available_slots = 10000000
+    available_slots = 5000000
     samtools = args.samtools
     tensor_file_path = args.tensor_fn
     bam_file_path = args.bam_fn
diff --git a/shared/param.py b/shared/param.py
index 51bb3bf..329a1c0 100644
--- a/shared/param.py
+++ b/shared/param.py
@@ -13,7 +13,7 @@
 
 # Model hyperparameters
 trainBatchSize = 10000
-predictBatchSize = 10000
+predictBatchSize = 1000
 initialLearningRate = 1e-3
 learningRateDecay = 0.1
 maxLearningRateSwitch = 3