In this small work, we solve a classification problem using pytorch. The idea is to compare the result obtained with keras. This project gives a samply way to do classification using pytorch. The result show that the learning in pytorch is fast Please, make sure you have dowloaded the data on
For Gluon dataset https://cernbox.cern.ch/index.php/s/9ZQG7aV1g8TH1ng
For quark dataset https://cernbox.cern.ch/index.php/s/8ep1MSKpsiLLcXH
and saved it in data/