cassianobecker / dnn

MIT License
4 stars 0 forks source link

fix CNN for MNIST #3

Closed cassianobecker closed 4 years ago

cassianobecker commented 4 years ago

Cleaned up the code and added \experiments\mnist\mnist_cnn_simple.py Please try to run the above script. This should require the minimum pytorch installation (over python 3.7)

According to the pytorch website, this should be achieved by:

conda install pytorch torchvision -c pytorch

(not sure how that translates to the spec file you mentioned)

For the near future, we will use packages: psutil sklearn

as well as many other, that we can add as need, provided we tell Harang (the package keeper :)