(self, saver, *args, **kwargs)
| 171 | |
| 172 | class IncrementalPyTorchPickler(pickle.Pickler): |
| 173 | def __init__(self, saver, *args, **kwargs): |
| 174 | super().__init__(*args, **kwargs) |
| 175 | self.storage_dtypes = {} |
| 176 | self.saver = saver |
| 177 | self.id_map = {} |
| 178 | |
| 179 | # this logic is taken from PyTorch 2.0+ torch/serialization.py |
| 180 | def persistent_id(self, obj): |