Swift port of ConvnetJS. [Work in progress]
Switch branches/tags
Nothing to show
Clone or download

README.md

ConvNetSwift

Swift port of ConvnetJS - a library for training Deep Learning models (mainly Neural Networks).

Updated to Swift 3.

Available NN layer types:

  • Fully Connected
  • Convolutional
  • Dropout
  • ReLU
  • Sigmoid
  • Maxout
  • Hyperbolic Tangent
  • Local Response Normalization
  • Max Pooling
  • Output: Softmax, Regression, SVM

Optimization methods ('trainers'/'solvers'):

  • Stochastic Gradient Descent
  • Adam
  • Adaptive Gradient
  • WindowGrad
  • AdaDelta
  • Nesterov’s Accelerated Gradient

Reinforcement learning:

  • Q-learning

Acceleration:

  • Accelerate framework
  • Metal

This is not a production-ready library. It is a personal research project and work in progress.