/Framework-Backpropagation

Neural Network Framework with backpropagation. The framework contains layers such as: Linear, ReLU, LeakyReLU, Sigmoid, SoftMax, Dropout, BatchNorm. Criterion: MSE, CrossEntropy. Optimizer: SGD. A model trained on the Mnist dataset was written.

Primary LanguageJupyter Notebook

Stargazers