| 97 | } |
| 98 | |
| 99 | int dataset::get_current_mini_batch_size() const |
| 100 | { |
| 101 | if (m_current_mini_batch_idx > (m_num_iterations_per_epoch - 1)) { |
| 102 | return 0; |
| 103 | } |
| 104 | else if (m_current_mini_batch_idx == (m_num_iterations_per_epoch - 1)) { |
| 105 | return m_last_mini_batch_size; |
| 106 | } |
| 107 | else { |
| 108 | return m_mini_batch_size; |
| 109 | } |
| 110 | } |
| 111 | |
| 112 | int dataset::get_next_position() const |
| 113 | { |
no outgoing calls
no test coverage detected