Source: https://medium.com/@UdacityINDIA/tensorflow-or-pytorch-the-force-is-strong-with-which-one-68226bb7dab4
1:
Tensorflow based on Theano developed by Google
PyTorch based on Torch developed by Facebook
2.
Tensorflow is static - design a computational graph first then run ML model.
PyTorch is dynamic - you can define/manipulate your graph on-the-go. Helpful while using variable length inputs in RNNs.
3.
PyTorch is a relatively new framework as compared to Tensorflow - so small community.
TensorBoard -enables visualizing your ML models directly in your browser.
PyTorch doesn’t have such a tool, although you can always use tools like Matplotlib
PyTorch is easier to learn and lighter to work with, and hence, is relatively better for passion projects and building rapid prototypes.
1:
Tensorflow based on Theano developed by Google
PyTorch based on Torch developed by Facebook
2.
Tensorflow is static - design a computational graph first then run ML model.
PyTorch is dynamic - you can define/manipulate your graph on-the-go. Helpful while using variable length inputs in RNNs.
3.
PyTorch is a relatively new framework as compared to Tensorflow - so small community.
TensorBoard -enables visualizing your ML models directly in your browser.
PyTorch doesn’t have such a tool, although you can always use tools like Matplotlib
PyTorch is easier to learn and lighter to work with, and hence, is relatively better for passion projects and building rapid prototypes.