A Simple Feed-Forward Image Classifier

PyTorch_Featured

Recently I dared to make first steps into the exciting and vast world of deep learning, which was a near impossible endeavor prior to finding a proper textbook to gently guide me into the field. While the text and the code examples found therein proved to be of great help, I decided to supplement those by writing my own examples (and extensively commenting them) and logging my progress by means of blog posts.

First yield of such practice is thus before you – a simple feed-forward neural network used to classify images into two classes, namely cats and fish.

Jupyter notebook of the model may be found on GitHub: https://tinyurl.com/yypx9ump

Also, the same notebook is available on Google Colab (where it can be tested it against the GPU available there): https://tinyurl.com/y3d2jfuu

Leave a Reply