[Feature Request] Make `SavedTensorDict` lazy
Open
@vmoens is already working on this.
Since Mar 17, 2022.
enhancement
- Dominant language
- Python
- Stars
- 3.6k
- Forks
- 484
- Avg merge
- 1d 1h
- Merged PRs (30d)
- 207
Description
Nearly all ops in SavedTensorDict could be done lazily:
class SavedTensorDict(_TensorDict):
def register_op(self, op, args, kwargs):
self._op_register.append((op, args, kwargs))
def some_op(self, ..):
self.register_op('some_op', ...)
def _load(self):
td = load(..)
for op, args, kwarg in self._op_register:
td = getattr(td, op)(*args, **kwargs)
return td
def _save_td(self, td):
....
self_op_register = []
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.