Skip to main content

A lightweight PyTorch code wrapper for ML researchers

Project description

NNCore

A lightweight PyTorch code wrapper for ML researchers.

NNCore is a library that provides common functionalities for Machine Learning and Deep Learning researchers. This project aims at helping users focus more on science but not engineering during researches. The essential functionalities include but are not limited to:

  • Universal I/O APIs
  • Efficient implementations of layers and losses that are not included in PyTorch
  • Extended methods for distributed training
  • More powerful data loading techniques
  • An engine that can take over the whole training and testing process, with all the baby-sitting works (stage control, optimizer configuration, lr scheduling, checkpoint management, metrics & tensorboard writing, etc.) done automatically. See an example for details.

Note that some methods in the library work with PyTorch 1.6+, but the installation of PyTorch is not necessary.

Continuous Integration

Platform / Python Version 3.6 3.7 3.8 3.9
Ubuntu 16.04 Build Build Build Build
Ubuntu 18.04 Build Build Build Build
Ubuntu 20.04 Build Build Build Build
macOS 10.15 Build Build Build Build
Windows Build Build Build Build

Installation

You may install nncore directly from PyPI

pip install nncore

or manually from source

git clone https://github.com/yeliudev/nncore.git
cd nncore
pip install -e .

Getting Started

Please refer to our documentation for how to incorperate nncore into your projects.

Acknowledgements

This library is licensed under the MIT License. Part of the code in this project is modified from mmcv and fvcore with many thanks to the original authors.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

nncore-0.2.5.tar.gz (54.1 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page