Re:PyTorch vs. TensorFlow in Academic Papers
by p1esk on Sunday March 13, 2022 @08:52PM
(#30667349)
Jax might be faster than Pytorch, I don’t know. I’m talking about TF. When I switched from TF to Pytorch 3 years ago, I got no slowdown on any of computer vision models at the time. And I remember looking at a couple of independent benchmarks which also showed them to be roughly the same in speed.
[ Parent
| on HN ]
Re:PyTorch vs. TensorFlow in Academic Papers
by brutus1213 on Tuesday March 15, 2022 @02:39PM
(#30689188)
It really depends on what your model is doing. For a time, sequence models were easier to do with Pytorch than TF (due to control flow). On the efficiency side, for vanilla CV models, I also did not observe major differences last time I looked, but when I started to do lots of things in parallel, multi-gpu training, heavy data augmentation, I think TF has some well-engineered capabilities that are not matched yet.