Published in


Image Classifier using PyTorch and Keras

Train a CNN on MNIST Dataset using Keras and PyTorch

Here you will learn to create a CNN model with similar architecture using Keras and PyTorch on MNIST dataset

Features of Keras and PyTorch.


  • Keras is a simpler, concise deep learning API written in Python that runs on TensorFlow's machine learning platform.
  • It enables fast experimentation.
  • Keras provides abstractions and…



Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store