Implementing Darknet19 from scratch

Hi, all –

Here’s a gist of my implementation of Darknet19 that classifies the MNIST dataset. I modeled the experiment off of course-v3’s Lesson 7 notebook because I found a need to better understand pytorch,, and the underlying theory. I ran a whole bunch of experiments and left only the most relevant ones in the notebook to keep the information straight-forward.

You’ll see that my implementation achieves 99.6% accuracy on MNIST in 12 epochs, clocking in at under 2mins training time on an NVIDIA Tesla P100.

Let me know what you think!


Hey @amqdn - just wanted to say thanks for this, this helped me a lot with my own work! I didn’t really understand how to create a model on my own until this.

1 Like