From a6adcf3d0df021696ced665799f76abaf542d543 Mon Sep 17 00:00:00 2001 From: "Brian C. Van Essen" Date: Fri, 12 Aug 2016 23:35:47 -0700 Subject: [PATCH] Fixed a bug where a parameter was errantly removed from the standard distributed input layer. --- src/layers/lbann_input_layer_distributed_minibatch.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/lbann_input_layer_distributed_minibatch.cpp b/src/layers/lbann_input_layer_distributed_minibatch.cpp index 5134ae72780..4a77c165f88 100644 --- a/src/layers/lbann_input_layer_distributed_minibatch.cpp +++ b/src/layers/lbann_input_layer_distributed_minibatch.cpp @@ -43,7 +43,7 @@ lbann::input_layer_distributed_minibatch::input_layer_distributed_minibatch(lban void lbann::input_layer_distributed_minibatch::setup(int num_prev_neurons) { if(io_layer::m_data_sets_span_models) { - io_layer::setup_data_readers(Layer::comm->get_num_models() * Layer::m_mini_batch_size, + io_layer::setup_data_readers(0, Layer::comm->get_num_models() * Layer::m_mini_batch_size, Layer::comm->get_model_rank() * Layer::m_mini_batch_size); }else { io_layer::setup_data_readers(0, m_mini_batch_size);