Viewing a single comment thread. View all comments

ContributionWild5778 OP t1_j5g4ck6 wrote

Reply to comment by OutrageousSundae8270 in Tensorflow or Pytorch by ContributionWild5778

Oh, okay. Thanks a lot! The reason I don't feel like moving on to pytorch is because I am a bit comfortable in TF now. And I don't have the energy or time to learn a different framework right now.

1

OutrageousSundae8270 t1_j5g51tx wrote

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.

1