From 72d73fc6a998b92e7a036ac934972e80bf74ee73 Mon Sep 17 00:00:00 2001 From: "david.seb.fischer" Date: Sun, 25 Aug 2019 18:57:13 +0200 Subject: [PATCH] changed default estimator batch size to 512 --- batchglm/train/tf/glm_beta/estimator.py | 2 +- batchglm/train/tf/glm_nb/estimator.py | 2 +- batchglm/train/tf/glm_norm/estimator.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/batchglm/train/tf/glm_beta/estimator.py b/batchglm/train/tf/glm_beta/estimator.py index 079f7174..fec6d10a 100644 --- 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 index afcf3298..a9b56650 100644 --- 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 index 531850ab..75c12e71 100644 --- 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",