Method get_next_position

src/utils/dataset.cpp:112–123

}

int dataset::get_next_position() const
{
  /// If the next mini-batch for this rank is going to be the last
  /// mini-batch, take the proper (possibly reduced) step to
  /// setup for the last mini-batch
  if (m_current_mini_batch_idx == (m_num_iterations_per_epoch - 1)) {
    return m_current_pos + m_stride_to_last_mini_batch;
  }
  else {
    return m_current_pos + m_stride_to_next_mini_batch;
  }
}

void dataset::set_mini_batch_size(const int s) { m_mini_batch_size = s; }