Submitted by ContributionWild5778 t3_10im3z3 in deeplearning
OutrageousSundae8270 t1_j5g51tx wrote
Reply to comment by ContributionWild5778 in Tensorflow or Pytorch by ContributionWild5778
I'm know the TensorFlow API really well considering I've been using it for about 2 years now.
I've used PyTorch before as well, its really not hard to learn the other once you're good at one of them.
PyTorch is a little more verbose than TensorFlow , e.g. TensorFlow has a default training loop that can be started by invoking the fit method where as in PyTorch you write the training loop yourself regardless, you always have to explicitly call methods for AD/Back propagation as well in PyTorch.
Viewing a single comment thread. View all comments