(self, name: str, t: torch.Tensor)
| 179 | ) |
| 180 | |
| 181 | def write_tensor(self, name: str, t: torch.Tensor) -> None: |
| 182 | storage = t.untyped_storage() |
| 183 | h = self.write_storage(storage) |
| 184 | # TODO: Support more advanced snapshotting of requires_grad/grad/etc |
| 185 | d, f = os.path.split(name) |
| 186 | payload = self.compute_tensor_metadata(t, h=h) |
| 187 | subfolder = os.path.join(self.loc, "tensors", d) |
| 188 | os.makedirs(subfolder, exist_ok=True) |
| 189 | torch.save(payload, os.path.join(subfolder, f)) |
| 190 | |
| 191 | |
| 192 | class ContentStoreReader: |