Skip to content
C++ framework for deep learning
C++ C Objective-C Cuda Makefile
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
src
test
.gitignore
LICENCE
Makefile
README.md
main.cpp
saga.h
saga.mk

README.md

Saga - A small C++ framework for deep learning.

This project is still very much work in progress.

Features

  • Relies exclusively on NVIDIA's cuDNN and cuda libraries. Ie, this does not work without an NVIDIA GPU.

  • FP32 and FP16 inference and training mode.

  • Convolution, BatchNorm, Fully-Connected, Dropout -layers

  • Adam optimizer with mixed precision training.

  • Can load (some) ONNX models

TODO

  • Scaling of gradients when using mixed precision learning.

  • Load more ONNX models

Other

Why?

I wanted to get a better understanding of how deep neural nets work.

What about the name?

The etymology of the name Sága is generally held to be connected to the Old Norse verb sjá, meaning "to see" (from Proto-Germanic *sehwan).

https://en.wikipedia.org/wiki/S%C3%A1ga_and_S%C3%B6kkvabekkr

You can’t perform that action at this time.