scaelles / OSVOS-TensorFlow

One-Shot Video Object Segmentation
http://vision.ee.ethz.ch/~cvlsegmentation/osvos/
GNU General Public License v3.0
438 stars 132 forks source link

Time comparison between PyTorch and TensorFlow #21

Closed irwinherrmann closed 6 years ago

irwinherrmann commented 6 years ago

So I did some time benchmarks and found that about 10 epochs takes the following amount of time:

Env: OS: Ubuntu 16.04 GPU: Dual 1080 TIs with HB SLI bridge

TensorFlow 1.9: 0.85 seconds PyTorch 0.5: 2.32 seconds

Is this consistent with what you've observed? I'm trying to figure out which of the following is true: 1) this is a side effect of my installation; 2) the TensorFlow version is much more optimized; 3) there is an underlying side effect that makes TensorFlow faster than PyTorch for this network.

scaelles commented 6 years ago

We can continue the discussion in the Pytorch repository.