Tensorflow vs PyTorch

This repository could be refered if you are looking a solution for one of following issues:

• PyTorch and Tensorflow equivalent models for MNIST dataset, both training and prediction
• Convert CNN or LSTM models between PyTorch and Tensorflow.
• PyTorch equivalent to tensorflow's model.summary()
• Deciding the LSTM parameters using PyTorch

Table of Contents

• CNN using TensorFlow
• CNN using PyTorch
• LSTM using Tensorflow
• LSTM using PyTorch