pytorch-learn

pytorch-learn is a lightweight deep learning framework for general use cases based on PyTorch.

Installation

pytorch-learn requires Python 3.6 or higher.

Pre-Installing PyTorch

pytorch-learn requires pytorch>=1.8.0. Please refer to PyTorch, and it is highly recommended to pre-install PyTorch with conda.

pip installation

After installing PyTorch, installation of pytorch-learn would be rather easy:

If you pre-installed PyTorch with conda, remember to activate the corresponding environment!

git clone https://github.com/pku-aiic/pytorch-learn.git
cd pytorch-learn
pip install -e .

Examples

Please visit the examples folder for more details.