changed default estimator batch size to 512

theislab · Aug 25, 2019 · 72d73fc · 72d73fc
1 parent 5ba171a
commit 72d73fc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/batchglm/train/tf/glm_beta/estimator.py b/batchglm/train/tf/glm_beta/estimator.py
@@ -20,7 +20,7 @@ class Estimator(TFEstimatorGLM, ProcessModel):
     def __init__(
             self,
             input_data: InputDataGLM,
-            batch_size: int = 500,
+            batch_size: int = 512,
             graph: tf.Graph = None,
             init_model: Model = None,
             init_a: Union[np.ndarray, str] = "AUTO",

diff --git a/batchglm/train/tf/glm_nb/estimator.py b/batchglm/train/tf/glm_nb/estimator.py
@@ -20,7 +20,7 @@ class Estimator(TFEstimatorGLM, ProcessModel):
     def __init__(
             self,
             input_data: InputDataGLM,
-            batch_size: int = 500,
+            batch_size: int = 512,
             graph: tf.Graph = None,
             init_model: Model = None,
             init_a: Union[np.ndarray, str] = "AUTO",

diff --git a/batchglm/train/tf/glm_norm/estimator.py b/batchglm/train/tf/glm_norm/estimator.py
@@ -22,7 +22,7 @@ class Estimator(TFEstimatorGLM, ProcessModel):
     def __init__(
             self,
             input_data: InputDataGLM,
-            batch_size: int = 500,
+            batch_size: int = 512,
             graph: tf.Graph = None,
             init_model: Model = None,
             init_a: Union[np.ndarray, str] = "AUTO",