Submitted by aleguida t3_yde1q8 in MachineLearning
seba07 t1_ittpya9 wrote
Reply to comment by TiredOldCrow in [D] Tensorflow learning differently than Pytorch by aleguida
Or otherwise don't use a pre-trained network for this test. Pytorch randomness shouldn't be better than Tensorflows.
aleguida OP t1_itvgiu8 wrote
Thanks for the feedback. I tried retraining everything from scratch without downloading any pretrained weights. here is the colab links update.
- TF: https://colab.research.google.com/drive/1O6qzopiFzK5tDmLQAzLKmNoNaEMDc4Ze?usp=sharing
- PYTORCH: https://colab.research.google.com/drive/1g-1CEpzmWJi9xOiZHzvDSv_-eDlXZO9u?usp=sharing
​
While Pytorch is learning something, Tf is not learning anything. This is actually quite confusing as I used tf.Keras to minimize any possible error on my part. I will try to build the same network from scratch in both frameworks next
[deleted] t1_itzwb38 wrote
[deleted]
Viewing a single comment thread. View all comments