Update avaiable_slots and prediction batch size

- prevent using too much memory
HKU-BAL · Dec 5, 2019 · a2213ad · a2213ad
1 parent 7c2b41c
commit a2213ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -13,7 +13,7 @@
 
 # Model hyperparameters
 trainBatchSize = 10000
-predictBatchSize = 10000
+predictBatchSize = 1000
 initialLearningRate = 1e-3
 learningRateDecay = 0.1
 maxLearningRateSwitch = 3