(self, saver, *args, **kwargs)
| 308 | |
| 309 | class IncrementalPyTorchPickler(pickle.Pickler): |
| 310 | def __init__(self, saver, *args, **kwargs): |
| 311 | super().__init__(*args, **kwargs) |
| 312 | self.storage_dtypes = {} |
| 313 | self.saver = saver |
| 314 | self.id_map = {} |
| 315 | |
| 316 | # this logic is taken from PyTorch 2.0+ torch/serialization.py |
| 317 | def persistent_id(self, obj): |