Skip to content

Commit 7440444

Browse files
authored
Update README.md
1 parent 80ca193 commit 7440444

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

README.md

+2
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ This repo contains all the code and figures that we produced during our project.
44

55
## Abstract
66

7+
In this project, we implemented 4 optimizers algorithms: SGD, SGD with momentum, RMSprop and ADAM and investigated the impact of them as well as its learning rate for solving a classification problem on multiple datasets: CIFAR10, CIFAR100, MNIST, FashionMNIST. We implemented the testing on the augmented data by Gaussian, flip and affine transforms in order to demonstrate the generalization capacities of optimizers. The behaviour analysis seem to show that momentum helps in generalization, which is also proven by the best accuracy reached by MSGD.
8+
79
## Requirements
810
The following packages are needed in order to run our code without error:
911
- python >= 3.8

0 commit comments

Comments
 (0)