Fixed a bug where a parameter was errantly removed from the standard …

…distributed input layer.
LLNL · Aug 13, 2016 · a6adcf3 · a6adcf3
1 parent 258d611
commit a6adcf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);