/TorchTree

Tree structure based on Pytorch's nn.Module machinery

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

TorchTree

A Python Tree library based on PyTorch nn.Module machinery. Allows to store data where modules are nodes and they contain any amount of parameters.
Provides Directory Tree class which generates a tree from a root directory.