DEV Community

Cover image for Why I Prefer PyTorch Over TensorFlow: Same Reason I Prefer Linux Over Windows
Amit jha
Amit jha

Posted on

Why I Prefer PyTorch Over TensorFlow: Same Reason I Prefer Linux Over Windows

Hey everyone as we kick start this series I would like to shed some light on why I prefer PyTorch over TensorFlow and why you should too.

Alt Text

As a deep learning practitioner I like (or require) custom architectures for my projects or research papers which can be difficult to build using TensorFlow because it’s implementation is not very friendly (non-pythonic). Hence, building a custom architecture in TensorFlow is time-consuming.

Additionally, unless you are using core TensorFlow you don’t have much control over anything other than the number of layers in a model and their configurations.

PyTorch, on the other hand, gives you complete control over how you want to train your model and what should be the underlying architecture. In fact, it uses OOP for building deep learning model and the training loop is an ACTUALLY python for loop!!

This gives you full control over how the model is trained and provides you with all the optimizations that libraries like TensorFlow provide.

Now that I’ve explained why I prefer PyTorch over TensorFlow we are ready to explore deep learning.

Cheers,
Jha

This series will also be available on Medium

Some of my other blog posts that you may like


Top comments (2)

Collapse
 
polmonroig profile image
Pol Monroig Company

So true... PyTorch just feels better.

Collapse
 
jha profile image
Amit jha

I agree. It's very pythonic. I love it.