Train, test, debug and optimize PyTorch models
Project description
Purpose
The aim of this library is to simplify the process of building, optimizing, testing and debugging deep learning models using PyTorch as well as providing implementations of some of the latest research papers. Extensibility is kept in mind so that it is easy to customize the framework for your particular needs.
Some key features of the framework:
Easy to use, flexible and extensible API to build simple & complex models
Model debugging (e.g., activation statistics of each layer, gradient norm for each layer, embedding visualization)
Model understanding and result analysis (e.g., attention maps, confusion matrix, ROC curves, model comparisons, errors)
Support hyper-parameter optimization (random search, hyperband) and analysis
Architecture learning (DARTS & evolutionary algorithms)
Keep track of the results for retrospective analysis and model selection
Requirements
Linux/Windows
Python >= 3.6
PyTorch >= 1.0
Installation / Usage
To install use pip:
$ pip install trw
Or clone the repo:
$ git clone https://github.com/civodlu/trw.git
$ python setup.py install
Documentation
The documentation can be found at ReadTheDocs.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.