TensorNet is a high-level deep learning library built on top of PyTorch.
You can use pip to install tensornet
pip install torch-tensornet
If you want to get the latest version of the code before it is released on PyPI you can install the library from GitHub
pip install git+https://github.com/shan18/TensorNet.git#egg=torch-tensornet
Documentation is available at: https://tensornet.readthedocs.io/en/latest/
For examples on how to use TensorNet, refer to the examples directory.
TensorNet has the following third-party dependencies
- torch
- torchvision
- torchsummary
- tqdm
- matplotlib
- albumentations
- opencv-python
If you need any help or want to report a bug, raise an issue in the repo.
