neural-pipeline 0.0.17

Neural Pipeline based on PyTorch 0.4.1 and designed to standardize and facilitate the training process

Neural networks training pipeline based on PyTorch 0.4.1 and designed to standardize and facilitate the training process.

It's contains:

• Checkpoints management and train process resuming
• Metrics processing and visualisation by builtins (tensorboard) or custom monitors
• Training best practices (e.g. learning rate decaying and hard negative mining)

Installation:

pip install neural-pipeline

For builtin module using install:

pip install tensorboardX

Install latest version before it's published on PyPi

pip install -U git+https://github.com/toodef/neural-pipeline

Getting started:

Documentation

See the full documentation there

See the examples

• MNIST classification example - notebook, file
• Segmentation example - notebook, file