| 208 | return (arg,) |
| 209 | |
| 210 | def _get_batch(self, epoch_idx): |
| 211 | try: |
| 212 | if self._batch: |
| 213 | callback_out = self._callback(*self._callback_args(None, epoch_idx)) |
| 214 | else: |
| 215 | callback_out = [ |
| 216 | self._callback(*self._callback_args(i, epoch_idx)) |
| 217 | for i in range(self._batch_size) |
| 218 | ] |
| 219 | self._current_sample += self._batch_size |
| 220 | self._current_iter += 1 |
| 221 | except StopIteration: |
| 222 | self._current_iter = 0 |
| 223 | self._current_sample = 0 |
| 224 | raise |
| 225 | return callback_out |
| 226 | |
| 227 | def _feed_input(self, data, kwargs): |
| 228 | if self._callback is not None: |