(self)
| 307 | return batch |
| 308 | |
| 309 | def __getstate__(self): |
| 310 | # TODO: add limited pickling support for sharing an iterator |
| 311 | # across multiple threads for HOGWILD. |
| 312 | # Probably the best way to do this is by moving the sample pushing |
| 313 | # to a separate thread and then just sharing the data queue |
| 314 | # but signalling the end is tricky without a non-blocking API |
| 315 | raise NotImplementedError("DataLoaderIterator cannot be pickled") |
| 316 | |
| 317 | def _shutdown_workers(self): |
| 318 | try: |
nothing calls this directly
no outgoing calls
no test coverage detected