| 84 | } |
| 85 | |
| 86 | int dataset::get_next_mini_batch_size() const |
| 87 | { |
| 88 | if (m_current_mini_batch_idx + 1 > (m_num_iterations_per_epoch - 1)) { |
| 89 | return 0; |
| 90 | } |
| 91 | else if (m_current_mini_batch_idx + 1 == (m_num_iterations_per_epoch - 1)) { |
| 92 | return m_last_mini_batch_size; |
| 93 | } |
| 94 | else { |
| 95 | return m_mini_batch_size; |
| 96 | } |
| 97 | } |
| 98 | |
| 99 | int dataset::get_current_mini_batch_size() const |
| 100 | { |
no outgoing calls
no test coverage detected