Choose wisely

Submitted by nickpngc t3_11t4c9c in deeplearning

Hello everyone, I am building my firsts deep-learning based projects and i just noticed that pytorch 2.0 is officially available. I started to learn tensorflow a while ago, but i have heard that pytorch is one of the most popular DL frameworks out there besides tf. Which one you guys prefer and why?

View Poll

2

Comments

You must log in or register to comment.

perrohunter t1_jchduna wrote

I’ve been a fervent user of TensorFlow since 2016 but we deduced to ditch TensorFlow at my company in favor of PyTorch because the community support and commitment from the PyTorch team is just way better

11

codeinassembly t1_jcj1cjd wrote

I wanted to modify model training to test a hypothesis. I chose TensorFlow. It may present a minor improvement in training speed (atleast before 2.0) but holy fuck was it not worth it with all the gotcha's.

It works well if you're okay with using their default pipeline. Attempt to setup anything custom and you'll be in for a lot of suffering unless you disable symbolic execution - at which point it's slower than PyTorch.

I tried to implement the same idea on PyTorch and was done in an hour. Use PyTorch; it's just better.

7

JacksOngoingPresence t1_jcjfaac wrote

+1
Default keras pipeline is fast, beginner friendly and great. As soon as something custom needs to be done - it's just painful if even possible.

Also the last time I checked (~a year ago) some features were heavily bugged ever since the introduction (literally for years). Like model.predict or tf.function cause memory leaks even with the "examples" code. That was the switching point for me.

3

tenbilliondollarsman t1_jcj8sc1 wrote

Same, i just started to learn tf 2.0 a couple weeks ago, i think from rn i should learn pytorch too gif

1