EduDL
Deep Learning library for educational purposes.
Initialize a new network using net.h. Building a dense network looks something like so:
Net bob(inputW); // Provide the input size here
bob.addDense(784, RELU);
bob.addDense(64, RELU);
bob.addDense(10, RELU);
bob.train_vec(trainSet, testSet, learning_rate, epochs, optimizer, batchSize, momentum, learning_decay);Currently supports
- Basic linear algebra
- Feedforward ANN
- Stochastic, Mini-batch Training
- Optimizers like Gradient Descent with Momentum, RMSprop
- Nonlinearities like RELU, Sigmoid, SoftMax
- Custom dataset operations like shuffling, splitting, batching, custom reader for MNIST, etc.
- BLIS Support