Skip to content
/ catalyst Public

Machine learning framework from scratch in C++. Watch out Tensorflow!

License

Notifications You must be signed in to change notification settings

one-2/catalyst

Repository files navigation

Stage 1

  • Basic neural network library
  • Main library is a Model parent and NeuralNetwork child
  • Supporting classes: Log, IO
  • Supporting namespaces: MatrixOps, Vis

Build commands: -> be in /build, then run cmake -DCMAKE_PREFIX_PATH=path/to/libtorch path/to/project cmake --build . --config Release

About

Machine learning framework from scratch in C++. Watch out Tensorflow!

Topics

Resources

License

Stars

Watchers

Forks