nnForge is a framework for training convolutional and fully-connected neural networks. It includes CPU and GPU (CUDA) backends. The schema of the networks is DAG (directed acyclic graph).
The framework has a number of layers defined:
Error functions available:
The library implements mini-batch Stochastic Gradient Descent training algorithm with optional momentum (vanilla or Nesterov), ADAM is supported as well.
nnForge is an open-source software distributed under the Apache License v2.0.
Download the latest version. Access all the releases along with release notes on GitHub.
The package contains nnForge framework as well as examples - applications using the framework.
The framework depends on Boost, OpenCV, and Protobuf.
If you want to use CUDA backend you will also need CUDA Toolkit installed and cuDNN v4 or newer installed.
nnForge is designed and implemented by Maxim Milakov.