neural-pipeline 0.0.19

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:

• Flexible and customizable training process
• Checkpoints management and train process resuming (source and target device independent)
• Metrics processing and visualisation by builtins (tensorboard, Matplotlib) or custom monitors
• Training best practices (e.g. learning rate decaying and hard negative mining)
• Metrics logging and comparision (DVC compatible)

Installation:

pip install neural-pipeline

For builtin module using install:

pip install tensorboardX matplotlib

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

Data flow scheme:

See the examples

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