changed default estimator batch size to 512

davidsebfischer · davidsebfischer · commit 72d73fc6a998 · 2019-08-25T18:57:13.000+02:00
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",
