(self)
| 1017 | """Similar to GlobalAsyncCheckpointManager but allows passing additional futures to be awaited |
| 1018 | while asynchronously serializing tensors. |
| 1019 | """ |
| 1020 | |
| 1021 | def __init__(self, *args, **kwargs): |
| 1022 | super().__init__(*args, **kwargs) |
| 1023 | self._loop = asyncio.new_event_loop() |
| 1024 | self._loop_thread = threading.Thread(target=self._loop.run_forever, daemon=True) |